Added String method to meet Classifier interface

sjwhitworth · Jul 17, 2019 · fadd963 · fadd963
1 parent bffc4a5
commit fadd963
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/naive/bernoulli_nb.go b/naive/bernoulli_nb.go
@@ -56,6 +56,10 @@ type BernoulliNBClassifier struct {
 	fitOn base.FixedDataGrid
 }
 
+func (nb *BernoulliNBClassifier) String() string {
+	return "BernoulliNBBClassifier"
+}
+
 func (nb *BernoulliNBClassifier) GetMetadata() base.ClassifierMetadataV1 {
 	return base.ClassifierMetadataV1{
 		FormatVersion:      1,