Method Cross validation

8d5773aa · Tags · 6c4536c4 · 8d5773aa
Commit 8d5773aa authored Aug 23, 2017 by Tags
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

news_classify_tag.py news_classify_tag.py +2 -2

No files found.
--- a/news_classify_tag.py
+++ b/news_classify_tag.py
@@ -132,8 +132,8 @@ class Classifier:

        self.vectorizer = CountVectorizer(min_df=2, stop_words='english', ngram_range=(1, 1))
        self.tfidf = TfidfTransformer()
-        # self.classifier = KNeighborsClassifier()
-        self.classifier = MLPClassifier()
+        self.classifier = KNeighborsClassifier()
+        # self.classifier = MLPClassifier()
        # self.classifier = Pipeline([
        #     ('vect', TfidfVectorizer(stop_words='english')),
        #     ('clf', MultinomialNB()),