python 保存训练好的预测模型 python如何加载训练好的模型pkl格式文件

Posted 2021-07-15 _刘文凯_

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