tsne fix

cpietsch · Nov 6, 2020 · 3cf1e75 · 3cf1e75
1 parent e7e11b7
commit 3cf1e75
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/similarity/tsne.js b/similarity/tsne.js
@@ -8,10 +8,10 @@ var path = require("path");
 const glob = require('glob-promise');
 const localPath = i => path.relative(process.cwd(), i)
 const argv = require('minimist')(process.argv.slice(2));
-const tf = require('@tensorflow/tfjs-node');
+const tf = require('@tensorflow/tfjs');
 // const tsne = require('@tensorflow/tfjs-tsne');
 const { createCanvas, loadImage } = require('canvas')
-const tsnejs = require('./lib/tsne-lib.js');
+const tsnejs = require('./lib/tsne.js');
 
 const canvas = createCanvas(224, 224)
 const ctx = canvas.getContext('2d')