aboutsummaryrefslogtreecommitdiff
path: root/src/server
diff options
context:
space:
mode:
authorab <abdullah_ahmed@brown.edu>2019-08-07 16:28:51 -0400
committerab <abdullah_ahmed@brown.edu>2019-08-07 16:28:51 -0400
commit17b27d3575d3f91f461262e5ad72a457238d198a (patch)
tree98e6c0a9f5a8f140ff6239a27ebc6f518c87280b /src/server
parent06b59a4ec2f1871846696da22928fc7d54ae02d6 (diff)
correlation matrix completed
Diffstat (limited to 'src/server')
-rw-r--r--src/server/Recommender.ts27
1 files changed, 26 insertions, 1 deletions
diff --git a/src/server/Recommender.ts b/src/server/Recommender.ts
index 3c71f3aa1..ea59703c3 100644
--- a/src/server/Recommender.ts
+++ b/src/server/Recommender.ts
@@ -1,4 +1,10 @@
+//import { Doc } from "../new_fields/Doc";
+//import { StrCast } from "../new_fields/Types";
+//import { List } from "../new_fields/List";
+//import { CognitiveServices } from "../client/cognitive_services/CognitiveServices";
+
var w2v = require('word2vec');
+var assert = require('assert');
export class Recommender {
@@ -10,6 +16,10 @@ export class Recommender {
Recommender.Instance = this;
}
+ /***
+ * Loads pre-trained model from word2vec
+ */
+
private loadModel(): Promise<any> {
let self = this;
return new Promise(res => {
@@ -20,6 +30,10 @@ export class Recommender {
});
}
+ /***
+ * Testing
+ */
+
public async testModel() {
if (!this._model) {
await this.loadModel();
@@ -33,6 +47,10 @@ export class Recommender {
}
}
+ /***
+ * Tests if instance exists
+ */
+
public async testInstance(text: string) {
if (!this._model) {
await this.loadModel();
@@ -40,14 +58,21 @@ export class Recommender {
console.log(text);
}
+ /***
+ * Uses model to convert words to vectors
+ */
+
public async vectorize(text: string[]) {
if (!this._model) {
await this.loadModel();
}
if (this._model) {
let word_vecs = this._model.getVectors(text);
- console.log(word_vecs[0]);
return word_vecs;
}
}
+
+
+
+
}