asreview.models.classifiers.NaiveBayesClassifier

class asreview.models.classifiers.NaiveBayesClassifier(alpha=3.822)[source]

Naive Bayes classifier (nb).

Naive Bayes classifier. Only works in combination with the asreview.models.feature_extraction.Tfidf feature extraction model. Though relatively simplistic, seems to work quite well on a wide range of datasets.

The naive Bayes classifier is an implementation based on the sklearn multinomial naive Bayes classifier.

Parameters: alpha (float, default=3.822) – Additive (Laplace/Lidstone) smoothing parameter (0 for no smoothing).

Attributes

`default_param`	Get the default parameters of the model.
`label`
`name`
`param`	Get the (assigned) parameters of the model.

Methods

`fit`(X, y)	Fit the model to the data.
`full_hyper_space`()	Get a hyperparameter space to use with hyperopt.
`hyper_space`()
`predict_proba`(X)	Get the inclusion probability for each sample.