diff --git a/Tutorial.ipynb b/Tutorial.ipynb index 862db3d..4d7e797 100644 --- a/Tutorial.ipynb +++ b/Tutorial.ipynb @@ -302,10 +302,10 @@ "source": [ "from sklearn.naive_bayes import GaussianNB\n", "\n", - "clf_gnb = DecisionTreeClassifier()\n", - "clf_gnb.fit(train_x_vectors, train_y)\n", + "clf_gnb = GaussianNB()\n", + "clf_gnb.fit(train_x_vectors.todense(), train_y)\n", "\n", - "clf_gnb.predict(test_x_vectors[0])\n" + "clf_gnb.predict(test_x_vectors.todense()[0])" ] }, { @@ -376,7 +376,7 @@ "# Mean Accuracy\n", "print(clf_svm.score(test_x_vectors, test_y))\n", "print(clf_dec.score(test_x_vectors, test_y))\n", - "print(clf_gnb.score(test_x_vectors, test_y))\n", + "print(clf_gnb.score(test_x_vectors.todense(), test_y))\n", "print(clf_log.score(test_x_vectors, test_y))" ] },