logml.models.registry.naive_bayes

Classes

BernoulliNBClassifierModel([params, logger])

Wrapper for sklearn.naive_bayes.BernoulliNB.

class logml.models.registry.naive_bayes.BernoulliNBClassifierModel(params: Optional[dict] = None, logger=None)

Bases: logml.models.base.BaseModel

Wrapper for sklearn.naive_bayes.BernoulliNB.

F_MODEL

alias of sklearn.naive_bayes.BernoulliNB

TASK = 'classification'

TAGS = ['naive_bayes']

FE_MODEL_ATTRIBUTE = None

DEFAULT_PARAMS = {'alpha': 1}

PARAMS_SPACE = {'alpha': [0, 0.1, 1], 'binarize': [0, 0.1, 1], 'fit_prior': [True, False]}