BernoulliNB

#include <Skigen/NaiveBayes>

template <typename Scalar = double>
class Skigen::BernoulliNB(alpha=1, force_alpha=true, binarize=0, fit_prior=true, class_prior=VectorType())

Naive Bayes classifier for multivariate Bernoulli models.

Like MultinomialNB, this classifier is suitable for discrete data. The difference is that while MultinomialNB works with occurrence counts, BernoulliNB is designed for binary/boolean features.

Mirrors sklearn.naive_bayes.BernoulliNB.

---

---

Attributes:

• class_count : VectorType

• class_log_prior : VectorType

• classes : const Eigen::VectorXi

• feature_count : MatrixType

• feature_log_prob : MatrixType

---

Methods

SKIGEN_PARAMS()

---

predict(X)

---

predict_proba(X)

---

predict_log_proba(X)

---

partial_fit(X, y, classes)

---