Use pure js bayes classifier

lib/classifier.js

@@ -0,0 +1,41 @@
+"use strict";
+
+/**
+ * Wrapper on bayes classifier.
+ *
+ * @module
+ */
+
+var bayes = require("bayes");
+
+module.exports = () => {
+
+    var classifier = bayes();
+    classifier.isTrained = false;
+
+    classifier.train = function () {
+        this.isTrained = true;
+    };
+
+    classifier.getClassifications = function (text) {
+        var tokens = this.tokenizer(text);
+        var frequencyTable = this.frequencyTable(tokens);
+
+        var result = [];
+        for (var category in this.categories) {
+
+            var item = { label: category, value: 0 };
+            result.push(item);
+
+            for (var [token, frequency] of Object.entries(frequencyTable)) {
+                var tokenProbability = this.tokenProbability(token, category);
+                item.value += frequency * tokenProbability;
+            };
+        };
+
+        result.sort((a, b) => b.value - a.value);
+        return result;
+    };
+
+    return classifier;
+};

lib/tools.js

@@ -12,16 +12,14 @@ require("colors");
 var _ = require("lodash");
 var expect = require("chai").expect;
 var highlight = require("cli-highlight").highlight;
-var natural = require("natural");
 var Testrail = require("testrail-api");
 var U = require("glace-utils");
 
+var classifier = require("./classifier")();
+
 var d = U.switchColor();
 var self = module.exports;
 
-var classifier = new natural.BayesClassifier();
-classifier.isTrained = false;
-
 var all_steps = [];
 all_steps.isFilled = false;
 
@@ -64,7 +62,7 @@ self.listSteps = (filter, namesOnly) => {
         if (!all_steps.isFilled) all_steps.push(tmp);
 
         if (doc && !classifier.isTrained) {
-            classifier.addDocument(doc, s);
+            classifier.learn(doc, s);
         }
 
         if (namesOnly) {
@@ -79,7 +77,6 @@ self.listSteps = (filter, namesOnly) => {
 
     if (!classifier.isTrained) {
         classifier.train();
-        classifier.isTrained = true;
     }
 
     if (filter && !namesOnly) {