diff --git a/src/MultinomialNaiveBayes.py b/src/MultinomialNaiveBayes.py index 67709e7..bd3174e 100644 --- a/src/MultinomialNaiveBayes.py +++ b/src/MultinomialNaiveBayes.py @@ -30,6 +30,8 @@ class MultinomialNaiveBayes: y = dataset['Label'] print(y[:12]) + + # bis hierhin stimmt noch alles... if sklearn_cv: cv = CountVectorizer() @@ -62,6 +64,9 @@ class MultinomialNaiveBayes: # use sklearn CountVectorizer # fit the training data and then return the matrix print('Title + Text von train') + + # aber ab hier stimmt was nicht. irgendwie rutschen da NaNs mit rein... + print(X[train]) print('Label von train')