OrgLion-ML/golearn
1
0
Fork 0
mirror of https://github.com/sjwhitworth/golearn.git synced 2025-04-26 13:49:14 +08:00
Code Issues Releases Wiki Activity

Added String method to meet Classifier interface

Browse Source
This commit is contained in:
Justin Judd 2019-07-17 08:42:30 -05:00
parent bffc4a52e6
commit fadd9630bf
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
naive/bernoulli_nb.go
View File

@ -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,