Change classifier to Kmean.

news_classify_tag.py

@@ -130,10 +130,12 @@ class Classifier:
         from sklearn.feature_extraction.text import CountVectorizer, TfidfVectorizer, TfidfTransformer
         from sklearn.pipeline import Pipeline
         from sklearn.naive_bayes import MultinomialNB
+        from sklearn.neighbors import KNeighborsClassifier
 
         self.vectorizer = CountVectorizer(min_df=2, stop_words='english', ngram_range=(1, 1))
         self.tfidf = TfidfTransformer()
-        self.classifier = MultinomialNB()
+        # self.classifier = MultinomialNB()
+        self.classifier = KNeighborsClassifier()
         # self.classifier = Pipeline([
         #     ('vect', TfidfVectorizer(stop_words='english')),
         #     ('clf', MultinomialNB()),