编程语言
首页 > 编程语言> > python 保存训练好的模型

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,xlf,模型,joblib,import
来源: https://blog.csdn.net/qq_24211837/article/details/118057447