fix(index): added confusionMatrix to the JSON methods (#11)

all-contributors · May 31, 2019 · 1b09acf · 1b09acf
1 parent 1fb5f33
commit 1b09acf
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/index.js b/src/index.js
@@ -227,6 +227,7 @@ class Learner {
     }
     if (this.macroAvg) json.macroAvg = this.macroAvg
     if (this.microAvg) json.microAvg = this.microAvg
+    if (this.confusionMatrix) json.confusionMatrix = this.confusionMatrix
     return json
   }
 
@@ -239,6 +240,7 @@ class Learner {
   static fromJSON(json) {
     const ALLOWED_PROPS = [
       'classifierBuilder',
+      'confusionMatrix',
       'trainSet',
       'testSet',
       'macroAvg',