naive_bayes

class CategoricalNaiveBayesVectorClassificationModel(pseudo_count=0.1)[source]

Bases: sensai.vector_model.VectorClassificationModel

Naive Bayes with categorical features

__init__(pseudo_count=0.1)

Parameters: pseudo_count – the count to add to each empirical count in order to avoid overfitting