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()

hyper_space()

predict_proba(X)

Get the inclusion probability for each sample.