From a1cef1cab4318b91705f93c559767fe45f87aebf Mon Sep 17 00:00:00 2001 From: Guy Coleman Date: Fri, 29 Nov 2024 16:39:07 +0000 Subject: [PATCH] Correct naive_bayes models docstring typos Fixes online-ml/river#1573 --- river/naive_bayes/complement.py | 4 ++-- river/naive_bayes/multinomial.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) 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