mathdatech/Python
1
0
Fork 0
mirror of https://github.com/TheAlgorithms/Python.git synced 2025-03-13 18:19:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed comments

Browse Source
This commit is contained in:
ricca 2023-10-03 20:18:09 +02:00
parent c375df23fd
commit 9fa8164458
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
machine_learning/multinomial_naive_bayes_classifier.py
View File

@ -1,8 +1,10 @@
"""
Implementation from scratch of a Multinomial Naive Bayes Classifier.
The algorithm is trained and tested on the twenty_newsgroup dataset from sklearn to perform text classification
The algorithm is trained and tested on the twenty_newsgroup dataset
from sklearn to perform text classification
Here the Wikipedia page to understand the theory behind this kind of probabilistic models:
Here the Wikipedia page to understand the theory behind this kind
of probabilistic models:
https://en.wikipedia.org/wiki/Naive_Bayes_classifier
"""
@ -28,7 +30,8 @@ def group_indices_by_target(targets: ArrayLike) -> dict:
Returns
----------
grouped_indices : dict of (label : list)
Maps each target label to the list of indices of the examples with that label
Maps each target label to the list of indices of the
examples with that label
Example
----------
@ -90,8 +93,8 @@ class MultinomialNBClassifier:
Example
----------
Let's test the function following an example taken from the documentation of the MultinomialNB model
from sklearn
Let's test the function following an example taken from the documentation
of the MultinomialNB model from sklearn
>>> rng = np.random.RandomState(1)
>>> data = rng.randint(5, size=(6, 100))
>>> data = sparse.csr_matrix(data)