在使用joblib保存SVR模型的时候,发现sklearn.externals.joblib飘红。
参考他们得知,
sklearn在sklearn 0.23后没有了joblib模块,如果要使用,直接安装joblib
pip install joblib
使用过程
以下是一个简单的使用例子
from sklearn import svm
from joblib import dump, load # 使用joblib保存svr训练的模型
X = [[940, 20, 3], [820, 30, 4], [820, 25, 2], [820, 30, 3], [820, 35, 2]] # 特征向量
y = [43, 43, 40,42, 42] # 对应的分数
clf = svm.SVR()
clf.fit(X, y)
# 保存模型
dump(clf, 'svr_test.joblib')
print('模型保存成功!')
保存后是一个.joblib格式的文件,可以调用模型
clf_loaded = load('svr_test.joblib')
result = clf_loaded.predict([[800, 30, 5]])
print(result) # [42.09005704]