learning_model 1.0.43 → 1.0.46

package/lib/utils/data_manager.ts

@@ -0,0 +1,32 @@
+import * as tf from '@tensorflow/tfjs';
+import { io } from '@tensorflow/tfjs-core';
+
+class DataManager {
+  private labelMap: { [key: string]: number } = {};
+  private labelIndex: number = 0;
+  private data: { label: number, values: number[] }[] = [];
+
+  async addData(label: string, values: number[]): Promise<void> {
+    if (!(label in this.labelMap)) {
+      this.labelMap[label] = this.labelIndex++;
+    }
+    const numericLabel = this.labelMap[label];
+    this.data.push({ label: numericLabel, values });
+  }
+
+  convertToTensors() {
+    const xs = this.data.map(d => d.values);
+    const ys = this.data.map(d => d.label);
+
+    const xsTensor = tf.tensor2d(xs, [xs.length, xs[0].length], 'float32');
+    const ysTensor = tf.tensor1d(ys, 'float32');
+
+    return { xs: xsTensor, ys: ysTensor };
+  }
+
+  getLabelMap() {
+    return this.labelMap;
+  }
+}
+
+export default DataManager;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "learning_model",
-  "version": "1.0.43",
+  "version": "1.0.46",
   "description": "learning model develop",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",