aboutsummaryrefslogtreecommitdiff
path: root/src/server/Recommender.ts
diff options
context:
space:
mode:
authorab <abdullah_ahmed@brown.edu>2019-08-05 18:15:36 -0400
committerab <abdullah_ahmed@brown.edu>2019-08-05 18:15:36 -0400
commita8a3af0193cc423fc7b0cff4ca9ff1a9074a7998 (patch)
tree539734b0d1706ef2955a2f4886ff2c5f6bd7a488 /src/server/Recommender.ts
parent38317d4a3687e02a727693b569dfc5980db81854 (diff)
promises etc
Diffstat (limited to 'src/server/Recommender.ts')
-rw-r--r--src/server/Recommender.ts33
1 files changed, 33 insertions, 0 deletions
diff --git a/src/server/Recommender.ts b/src/server/Recommender.ts
new file mode 100644
index 000000000..6d9ca6486
--- /dev/null
+++ b/src/server/Recommender.ts
@@ -0,0 +1,33 @@
+var w2v = require('word2vec');
+
+export class Recommender {
+
+ private _model: any;
+
+ constructor() {
+ console.log("creating recommender...");
+ }
+
+ public loadModel(): Promise<any> {
+ let self = this;
+ return new Promise(res => {
+ w2v.loadModel("./node_modules/word2vec/vectors.txt", function (err: any, model: any) {
+ console.log(err);
+ console.log(model);
+ self._model = model;
+ console.log(model.similarity('father', 'mother'));
+ res(model);
+ });
+ });
+ }
+
+ public testModel() {
+ if (this._model) {
+ let similarity = this._model.similarity('father', 'mother');
+ console.log(similarity);
+ }
+ else {
+ console.log("model not found :(");
+ }
+ }
+}