Merge branch 'online-ml:main' into rls_branch

online-ml · Nov 30, 2024 · 8d5a3c8 · 8d5a3c8
2 parents 8752ab4 + a1cef1c
commit 8d5a3c8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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