python 保存训练好的模型

2021/6/19 20:27:19

本文主要是介绍python 保存训练好的模型,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

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) # 模型使用


这篇关于python 保存训练好的模型的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!


扫一扫关注最新编程教程