Re-train the models and add a condition for Predict

Fixes #153
olivia-ai · Feb 25, 2021 · 109dca9 · 109dca9
1 parent 702fbe3
commit 109dca9
Show file tree

Hide file tree

Showing 10 changed files with 12 additions and 9 deletions.
diff --git a/analysis/sentence.go b/analysis/sentence.go
@@ -54,6 +54,9 @@ func (sentence Sentence) PredictTag(neuralNetwork network.Network) string {
 	// Enumerate the results with the intent tags
 	var resultsTag []Result
 	for i, result := range predict {
+		if i >= len(classes) {
+			continue
+		}
 		resultsTag = append(resultsTag, Result{classes[i], result})
 	}
 

diff --git a/res/locales/ca/training.json b/res/locales/ca/training.json
diff --git a/res/locales/de/training.json b/res/locales/de/training.json
diff --git a/res/locales/el/training.json b/res/locales/el/training.json
diff --git a/res/locales/en/training.json b/res/locales/en/training.json
diff --git a/res/locales/es/training.json b/res/locales/es/training.json
diff --git a/res/locales/fr/training.json b/res/locales/fr/training.json
diff --git a/res/locales/it/training.json b/res/locales/it/training.json
diff --git a/res/locales/nl/training.json b/res/locales/nl/training.json
diff --git a/res/locales/tr/training.json b/res/locales/tr/training.json