尝试2 methods
了两者都给出了相同的错误,请帮助我将其保存为泡菜文件,以便我以后可以使用该模型。
我希望service.exe
使用该模型进行创建,请建议一种最佳方法,因为我对数据科学比较陌生
result = classifier.predict(kyc_test_image)
kyc_training_set.class_indices
if result[0][0] == 1:
prediction = 'pan'
else:
prediction = 'driv'
kyc_model = classifier
import pickle
modelname = 'cnn_kyc'
outfile = open(modelname,'wb')
pickle.dump(kyc_model,outfile)
outfile.close()
TypeError Traceback (most recent call last)
<ipython-input-35-865fb4ce2e0b> in <module>
----> 1 pickle.dump(kyc_model,outfile)
2 outfile.close()
TypeError: can't pickle _thread._local objects
第二种方法
from sklearn.externals import joblib
joblib.dump(kyc_model, 'cnn_kyc.pkl')
cnn_from_joblib = joblib.load('cnn_kyc.pkl')
cnn_from_joblib.predict(kyc_test_image)