bug fix for smoothing

gnames · Jun 14, 2018 · e966c1e · e966c1e
1 parent bc9d9fa
commit e966c1e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/runner.go b/runner.go
@@ -141,12 +141,14 @@ func likelihood(nb *NaiveBayes, feature Featurer, label Labeler) float64 {
 	countFeature := nb.FeatureFreq[name][value][label]
 
 	countRest := (nb.FeatureTotal[name][value] - countFeature)
-	pFeature := countFeature / nb.LabelFreq[label]
 
 	// crude smoothing
 	if countFeature == 0 {
 		countFeature = smooth
 	}
+
+	pFeature := countFeature / nb.LabelFreq[label]
+
 	if countRest == 0 {
 		countRest = smooth
 	}