diff --git a/river/naive_bayes/complement.py b/river/naive_bayes/complement.py index 80598b25eb..0a02f3a208 100644 --- a/river/naive_bayes/complement.py +++ b/river/naive_bayes/complement.py @@ -28,8 +28,8 @@ class ComplementNB(base.BaseNB): Attributes ---------- - class_dist : proba.Multinomial - Class prior probability distribution. + class_counts : collections.Counter + Number of times each class has been seen. feature_counts : collections.defaultdict Total frequencies per feature and class. class_totals : collections.Counter diff --git a/river/naive_bayes/multinomial.py b/river/naive_bayes/multinomial.py index 97ce314d50..9041d2b9d8 100644 --- a/river/naive_bayes/multinomial.py +++ b/river/naive_bayes/multinomial.py @@ -29,8 +29,8 @@ class MultinomialNB(base.BaseNB): Attributes ---------- - class_dist : proba.Multinomial - Class prior probability distribution. + class_counts : collections.Counter + Number of times each class has been seen. feature_counts : collections.defaultdict Total frequencies per feature and class. class_totals : collections.Counter