aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorab <abdullah_ahmed@brown.edu>2019-10-17 11:57:58 -0400
committerab <abdullah_ahmed@brown.edu>2019-10-17 11:57:58 -0400
commit46f25c9a781783350a7c1d76eefb4e066b2cac83 (patch)
tree9d9a9bc8551a576d58dc0710d4a9c72a9f943a12
parent3fc0d321b3b3ee09955c35e0a7bb260153b013e1 (diff)
optimize tf
-rw-r--r--package.json4
-rw-r--r--src/client/cognitive_services/CognitiveServices.ts1
-rw-r--r--src/server/Recommender.ts1
3 files changed, 5 insertions, 1 deletions
diff --git a/package.json b/package.json
index fbed088fd..f138e7a79 100644
--- a/package.json
+++ b/package.json
@@ -58,6 +58,7 @@
"@hig/theme-data": "^2.3.3",
"@tensorflow-models/universal-sentence-encoder": "^1.2.0",
"@tensorflow/tfjs": "^1.2.9",
+ "@tensorflow/tfjs-node": "^1.2.11",
"@trendmicro/react-dropdown": "^1.3.0",
"@types/adm-zip": "^0.4.32",
"@types/animejs": "^2.0.2",
@@ -122,8 +123,8 @@
"@types/youtube": "0.0.38",
"adm-zip": "^0.4.13",
"archiver": "^3.0.3",
- "arxiv-api-node": "0.0.2",
"array-batcher": "^1.1.3",
+ "arxiv-api-node": "0.0.2",
"async": "^2.6.2",
"babel-runtime": "^6.26.0",
"bcrypt-nodejs": "0.0.3",
@@ -173,6 +174,7 @@
"mobx-utils": "^5.4.0",
"mongodb": "^3.1.13",
"mongoose": "^5.6.4",
+ "node-pre-gyp": "^0.13.0",
"node-sass": "^4.12.0",
"nodemailer": "^5.1.1",
"nodemon": "^1.18.10",
diff --git a/src/client/cognitive_services/CognitiveServices.ts b/src/client/cognitive_services/CognitiveServices.ts
index 48519f916..b0e9138a4 100644
--- a/src/client/cognitive_services/CognitiveServices.ts
+++ b/src/client/cognitive_services/CognitiveServices.ts
@@ -267,6 +267,7 @@ export namespace CognitiveServices {
export async function vectorize(keyterms: any, dataDoc: Doc, mainDoc: boolean = false) {
console.log("vectorizing...");
//keyterms = ["father", "king"];
+
let args = { method: 'POST', uri: Utils.prepend("/recommender"), body: { keyphrases: keyterms }, json: true };
await requestPromise.post(args).then(async (wordvecs) => {
if (wordvecs) {
diff --git a/src/server/Recommender.ts b/src/server/Recommender.ts
index d974d7ef6..aaed09999 100644
--- a/src/server/Recommender.ts
+++ b/src/server/Recommender.ts
@@ -9,6 +9,7 @@ var arxivapi = require('arxiv-api-node');
import requestPromise = require("request-promise");
import * as use from '@tensorflow-models/universal-sentence-encoder';
import { Tensor } from "@tensorflow/tfjs-core/dist/tensor";
+require('@tensorflow/tfjs-node');
//http://gnuwin32.sourceforge.net/packages/make.htm