diff options
| author | ab <abdullah_ahmed@brown.edu> | 2019-08-05 18:15:36 -0400 |
|---|---|---|
| committer | ab <abdullah_ahmed@brown.edu> | 2019-08-05 18:15:36 -0400 |
| commit | a8a3af0193cc423fc7b0cff4ca9ff1a9074a7998 (patch) | |
| tree | 539734b0d1706ef2955a2f4886ff2c5f6bd7a488 /src/server/Recommender.ts | |
| parent | 38317d4a3687e02a727693b569dfc5980db81854 (diff) | |
promises etc
Diffstat (limited to 'src/server/Recommender.ts')
| -rw-r--r-- | src/server/Recommender.ts | 33 |
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 :("); + } + } +} |
