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) # 模型使用
sklearn保存训练好的模型为pkl,再调用
最新推荐文章于 2024-07-16 15:12:22 发布