Correct naive_bayes models docstring typos

Fixes #1573
online-ml · Nov 29, 2024 · a1cef1c · a1cef1c
1 parent 0390346
commit a1cef1c
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