diff --git a/naive/bernoulli_nb.go b/naive/bernoulli_nb.go index e26a318..767a90a 100644 --- 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,