fix(knowledge): FT parameters

botpress · Dec 4, 2018 · 827124c · 827124c
1 parent 79a55d8
commit 827124c
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/modules/knowledge/src/backend/classifier.ts b/modules/knowledge/src/backend/classifier.ts
@@ -64,7 +64,17 @@ export class DocumentClassifier {
     }
 
     const ft = new FastTextWrapper(fullModelPath)
-    ft.train(trainFile, { method: 'supervised' })
+    ft.train(trainFile, {
+      method: 'supervised',
+      epoch: 10,
+      bucket: 100000,
+      minCount: 1,
+      minn: 3,
+      maxn: 6,
+      wordGram: 5,
+      dim: 100,
+      learningRate: 0.2
+    })
 
     this._ft = ft
     this._index = indexedSnippets