@thi.ng/tsne 0.1.0 → 0.1.1

package/CHANGELOG.md

@@ -1,6 +1,6 @@
 # Change Log
 
-- **Last updated**: 2025-02-19T20:59:58Z
+- **Last updated**: 2025-02-20T09:10:22Z
 - **Generator**: [thi.ng/monopub](https://thi.ng/monopub)
 
 All notable changes to this project will be documented in this file.
@@ -11,6 +11,12 @@ See [Conventional Commits](https://conventionalcommits.org/) for commit guidelin
 **Note:** Unlisted _patch_ versions only involve non-code or otherwise excluded changes
 and/or version bumps of transitive dependencies.
 
+### [0.1.1](https://github.com/thi-ng/umbrella/tree/@thi.ng/tsne@0.1.1) (2025-02-20)
+
+#### ⏱ Performance improvements
+
+- more efficient use of vector ops ([af07e27](https://github.com/thi-ng/umbrella/commit/af07e27))
+
 ## [0.1.0](https://github.com/thi-ng/umbrella/tree/@thi.ng/tsne@0.1.0) (2025-02-19)
 
 #### 🚀 Features

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@thi.ng/tsne",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "description": "Highly configurable t-SNE implementation for arbitrary dimensions",
   "type": "module",
   "module": "./index.js",
@@ -82,5 +82,5 @@
     "status": "alpha",
     "year": 2021
   },
-  "gitHead": "bee617702ac61d093465b967f8f973dc566faa6b\n"
+  "gitHead": "76bad4103057c3a3edbe02740bfe5f342ac8a6b5\n"
 }

package/tsne.d.ts

@@ -1,5 +1,5 @@
 import type { FloatArray } from "@thi.ng/api";
-import { type ReadonlyVec } from "@thi.ng/vectors";
+import { type DistanceFn, type ReadonlyVec, type VecOpVN, type VecOpVV } from "@thi.ng/vectors";
 import type { TSNEOpts } from "./api.js";
 export declare const DEFAULT_OPTS: TSNEOpts;
 export declare class TSNE {
@@ -13,6 +13,9 @@ export declare class TSNE {
     points: number[][];
     steps: number[][];
     gains: number[][];
+    opDist: DistanceFn;
+    opDivN: VecOpVN;
+    opSub: VecOpVV;
     constructor(points: ReadonlyVec[], opts?: Partial<TSNEOpts>);
     init(points: ReadonlyVec[]): void;
     update(): number;

package/tsne.js

@@ -10,13 +10,10 @@ import {
   sub,
   zeroes
 } from "@thi.ng/vectors";
-const $distSq = distSq.impl();
-const $divN = divN.impl();
-const $sub = sub.impl();
 const EPS = Number.EPSILON;
 const DEFAULT_OPTS = {
   rnd: SYSTEM,
-  dist: $distSq,
+  dist: distSq,
   perplexity: 10,
   rate: 100,
   eps: 1e-4,
@@ -47,6 +44,9 @@ class TSNE {
   points;
   steps;
   gains;
+  opDist;
+  opDivN;
+  opSub;
   constructor(points, opts = {}) {
     this.opts = { ...DEFAULT_OPTS, ...opts };
     this.init(points);
@@ -55,8 +55,11 @@ class TSNE {
     const opts = this.opts;
     const n = this.n = points.length;
     const dim = this.dim = points[0].length;
+    this.opDist = opts.dist === distSq ? distSq.impl(dim) : opts.dist;
+    this.opDivN = divN.impl(dim);
+    this.opSub = sub.impl(dim);
     this.p = initProbabilities(
-      pairwiseDistances(points, opts.dist),
+      pairwiseDistances(points, this.opDist),
       n,
       opts.perplexity,
       opts.eps,
@@ -71,7 +74,7 @@ class TSNE {
   }
   update() {
     if (++this.iter >= this.opts.maxIter) return 0;
-    const { n, dim, points, steps, gains } = this;
+    const { n, dim, points, steps, gains, opDivN, opSub } = this;
     const {
       rate,
       minGain,
@@ -104,13 +107,12 @@ class TSNE {
         ymean[d] += row[d];
       }
     }
-    $divN(null, ymean, n);
-    for (i = 0; i < n; i++) $sub(null, points[i], ymean);
+    opDivN(null, ymean, n);
+    for (i = 0; i < n; i++) opSub(null, points[i], ymean);
     return cost;
   }
   computeGradient() {
-    const { n, dim, points: y, p, q, qu } = this;
-    const { dist, gradientScale } = this.opts;
+    const { n, dim, points: y, p, q, qu, opDist } = this;
     let i, j, rowIdx, d;
     let rowI, rowJ;
     let qsum = 0;
@@ -118,7 +120,7 @@ class TSNE {
       rowIdx = i * n;
       rowI = y[i];
       for (j = i + 1; j < n; j++) {
-        d = 1 / (1 + dist(rowI, y[j]));
+        d = 1 / (1 + opDist(rowI, y[j]));
         qu[rowIdx + j] = d;
         qu[j * n + i] = d;
         qsum += 2 * d;
@@ -130,7 +132,7 @@ class TSNE {
     }
     let cost = 0;
     const gradient = new Array(n);
-    const gscale = tweenParam(gradientScale, this.iter);
+    const gscale = tweenParam(this.opts.gradientScale, this.iter);
     for (i = 0; i < n; i++) {
       rowIdx = i * n;
       rowI = y[i];