Added import model from JSON to handle multiple models (#11)

serendipious · Jun 29, 2016 · a60d381 · a60d381
1 parent 1943a3b
commit a60d381
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/lib/decision-tree.js b/lib/decision-tree.js
@@ -82,6 +82,13 @@ module.exports = (function() {
 
       return correct / total;
     },
+
+    /**
+     * Imports a previously saved model with the toJSON() method
+     */
+    import: function(json) {
+      model = json;
+    },
 
     /**
      * Returns JSON representation of trained model