BernoulliNBClassifier
BernoulliNBClassifier
A model type for constructing a Bernoulli naive Bayes classifier, based on MLJScikitLearnInterface.jl, and implementing the MLJ model interface.
From MLJ, the type can be imported using
BernoulliNBClassifier = @load BernoulliNBClassifier pkg=MLJScikitLearnInterface
Do model = BernoulliNBClassifier()
to construct an instance with default hyper-parameters. Provide keyword arguments to override hyper-parameter defaults, as in BernoulliNBClassifier(alpha=...)
.
Binomial naive bayes classifier. It is suitable for classification with binary features; features will be binarized based on the binarize
keyword (unless it's nothing
in which case the features are assumed to be binary).