DOC Ensures that CategoricalNB passes numpydoc validation (#20445)

j3nnn1 · glemaitre · web-flow · commit 23c74e9afb03 · 2021-07-06T11:08:11.000+02:00
Co-authored-by: Guillaume Lemaitre &lt;g.lemaitre58@gmail.com&gt;
diff --git a/maint_tools/test_docstrings.py b/maint_tools/test_docstrings.py
@@ -17,7 +17,6 @@
     "Birch",
     "CCA",
     "CalibratedClassifierCV",
-    "CategoricalNB",
     "ClassifierChain",
     "ColumnTransformer",
     "ComplementNB",
diff --git a/sklearn/naive_bayes.py b/sklearn/naive_bayes.py
@@ -1142,7 +1142,7 @@ def _joint_log_likelihood(self, X):
 
 
 class CategoricalNB(_BaseDiscreteNB):
-    """Naive Bayes classifier for categorical features
+    """Naive Bayes classifier for categorical features.
 
     The categorical Naive Bayes classifier is suitable for classification with
     discrete features that are categorically distributed. The categories of
@@ -1216,6 +1216,12 @@ class CategoricalNB(_BaseDiscreteNB):
 
         .. versionadded:: 0.24
 
+    See Also
+    --------
+    ComplementNB : Complement Naive Bayes classifier.
+    GaussianNB : Gaussian Naive Bayes.
+    MultinomialNB : Naive Bayes classifier for multinomial models.
+
     Examples
     --------
     >>> import numpy as np
