#-*- coding:utf-8 -*- from classify import Classify import numpy as np import sys reload(sys) sys.setdefaultencoding("utf8") if __name__ == "__main__": X_train = np.array([ u"我想听张学友的歌", u"周杰伦的龙卷风", u"鹿晗有什么歌好听", u"姚明打篮球好厉害", u"张继科会打乒乓球", u"詹姆士是体育明星" ]) Y_train = np.array([1, 1, 1, 2, 2, 2]) Test_data = [u"我想听薛之谦的演员", "邓亚萍是体育明星", "刘翔是体育明星"] Model = Classify() Model.load_W2V_Model("word2vec.model") Model.train(X_train, Y_train) Model.predict(Test_data) Model.save_NBmodel("NB.model") del Model NBmodel_test = Classify() NBmodel_test.load_NBmodel("NB.model") NBmodel_test.predict(Test_data) del NBmodel_test