保存SVM模型并调用

将训练好的模型保存为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")

本文调用的模型是识别植物叶片是否病害。

你可能感兴趣的:(图像处理)