文章标题

#导入保存模型工具包from sklearn.externals import joblib#导入操作系统包import os#定位到这个路径os.chdir("/Users/yeshuai/")#导入sklearn的svm算法from sklearn import svm#定义数据特征矩阵x = [[1,2,3], [2,3,4],[1,2,3], [2,3,4],[1,2,3], [2,3,4],[1,2,3], [2,3,4],[1,2,3], [2,3,4]]#定义数据目标值y = [0, 1,0, 1,0, 1,0, 1,0, 1]#定义svmclf = svm.SVC()#训练模型clf.fit(x, y)#定义测试矩阵test_x = [[5,2,3], [3,3,4],[1,2,3], [2,3,4],[1,2,3], [2,3,4],[1,2,3], [2,5,4],[1,2,3], [2,3,4]]#打印测试数据的预估值print(clf.predict(test_x))# 保存模型到/Users/yeshuai/train_model.pkljoblib.dump(clf, "train_model.pkl")#加载模型model=joblib.load("train_model.pkl")#使用模型print(yeshuai.predict(test_x))