将训练好的模型保存为pickle文件,有以下步骤:
1)以二进制的方式打开文件:
file = open("G:/Download_Data_bishe/save_model/model.pickle", "wb")
2)然后将模型写入文件:pickle.dump(classifier, file)
3)最后关闭文件:file.close()
要想调用保存好的模型,有以下思路:
1) 以读二进制的方式打开文件
file = open("G:/Download_Data_bishe/save_model/model.pickle", "rb")
2)把模型从文件中读取出来
svm_model = pickle.load(file)
3)关闭文件
file.close()
4)用读入的模型进行预测
pred= svm_model.predict(X)
if pred==0:
print("病害")
if pred==1:
print("健康")
print("\n")
本文调用的模型是识别植物叶片是否病害。