mathdatech/Python
1
0
Fork 0
mirror of https://github.com/TheAlgorithms/Python.git synced 2025-02-07 01:50:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

Comments added

Browse Source
This commit is contained in:
ricca 2023-10-03 18:46:20 +02:00
parent 37184e21de
commit 5c4e412a35
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
machine_learning/multinomial_naive_bayes_classifier.py
View File

@ -1,5 +1,6 @@
"""
Implementation from scratch of a basic Multinomial Naive Bayes classifier for text classification.
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
"""
@ -54,7 +55,7 @@ class MultinomialNBClassifier:
self._check_X(X)
if X.shape[0] != len(y):
raise ValueError(
"The expected dimension for array y is (" + str(X.shape[0]) + ",), but got (" + str(len(y)) + ",)")
"The expected shape for array y is (" + str(X.shape[0]) + ",), but got (" + str(len(y)) + ",)")
def fit(self, X, y):
"""