Update classificator.js

Fix for when `"totalDocuments":0` or `"vocabularySize":0`
Wozacosta · Jul 6, 2021 · 19d6850 · 19d6850
1 parent 39cc888
commit 19d6850
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/classificator.js b/lib/classificator.js
@@ -55,7 +55,7 @@ module.exports.fromJson = (jsonStrOrObject) => {
 
   // override the classifier's state
   STATE_KEYS.forEach((k) => {
-    if (!parameters[k]) {
+    if (typeof parameters[k] === 'undefined') {
       throw new Error(
         `Naivebayes.fromJson: JSON string is missing an expected property: [${k}].`
       );