Merge pull request #13 from digitaldogsbody/master

Add ability for the TSNE script to use the tfjs-node library
cpietsch · Jun 19, 2022 · c1bb9da · c1bb9da
2 parents 6e0e941 + 5436a04
commit c1bb9da
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/similarity/package.json b/similarity/package.json
@@ -1,6 +1,6 @@
 {
   "name": "vikus-viewer-tsne",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "description": "tsne preprocessing script for vikus-viewer",
   "author": "Christopher Pietsch <cpietsch@gmail.com>",
   "private": true,

diff --git a/similarity/tsne.js b/similarity/tsne.js
@@ -8,7 +8,7 @@ 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');
+// const tf = require('@tensorflow/tfjs');
 // const tsne = require('@tensorflow/tfjs-tsne');
 const { createCanvas, loadImage } = require('canvas')
 const tsnejs = require('./lib/tsne.js');
@@ -20,6 +20,15 @@ console.log('starting with', process.argv);
 
 const inputPath = argv.i;
 const inputFormat = argv.f || 'jpg';
+const useTfjsnode = argv.t || false;
+
+var tf
+if(useTfjsnode) {
+  tf = require('@tensorflow/tfjs-node');
+}
+else {
+  tf = require('@tensorflow/tfjs');
+}
 
 const saveCsv = async (data, filename) => { 
   const csv = d3.csvFormat(data) 
@@ -180,4 +189,4 @@ run()
 //   // the embedded data.
 //   const coordinates = tsneOpt.coordinates();
 //   coordinates.print();
-// })
+// })