Add predict function to SNE

ishii-norimi · ishii-norimi · commit b660f2ac3f19 · 2022-01-01T12:06:34.000+09:00
diff --git a/lib/model/tsne.js b/lib/model/tsne.js
@@ -129,6 +129,14 @@ export class SNE {
 
 		return new Matrix(n, d, this._y).toArray()
 	}
+
+	/**
+	 * Returns reduced values.
+	 * @returns {Array<Array<number>>}
+	 */
+	predict() {
+		return Matrix.fromArray(this._y).toArray()
+	}
 }
 
 /**

Original file line number	Diff line number	Diff line change
`@@ -129,6 +129,14 @@ export class SNE {`
`129`	`129`
`130`	`130`	`return new Matrix(n, d, this._y).toArray()`
`131`	`131`	`}`
	`132`	`+`
	`133`	`+ /**`
	`134`	`+ * Returns reduced values.`
	`135`	`+ * @returns {Array<Array<number>>}`
	`136`	`+ */`
	`137`	`+ predict() {`
	`138`	`+ return Matrix.fromArray(this._y).toArray()`
	`139`	`+ }`
`132`	`140`	`}`
`133`	`141`
`134`	`142`	`/**`