ptyhon如何保存训练好的模型,python储存训练好的模型 python如何读取存储的模型
(以svm为例)
如下所示:
import numpy as np
from sklearn import svm
x_data = ... # 写自己的
y_label = ... # 写自己的
x_test = ... # 写自己的
xlf = svm.SVC(C=1,gamma=1,kernel='rbf',probability=True) # 构建模型
xlf.fit(x_data,y_label) # 训练模型
from sklearn.externals import joblib
joblib.dump(xlf, 'modle.pkl') # 储存模型 .dump(模型,‘模型名')
xlf_lo = joblib.load('modle.pkl') # 读取模型
y_test = xlf_lo.predict(x_test) # 模型使用
如果导入joblib包时遇到错误,请看:https://blog.csdn.net/qq_24211837/article/details/118057600