记录使用 sklearn保存模型 From Morvan

最新推荐文章于 2024-11-05 15:32:33 发布

不含硫jun

最新推荐文章于 2024-11-05 15:32:33 发布

阅读量76

点赞数

文章标签： sklearn python

本文链接：https://blog.csdn.net/sun1311523821/article/details/121135979

版权

from sklearn import svm
from sklearn import datasets

clf = svm.SVC()
iris = datasets.load_iris()
X, y = iris.data, iris.target
clf.fit(X, y)

# method 1: pickle
import pickle
# save
with open('save/clf.pickle', 'wb') as f:
    pickle.dump(clf, f)
# restore
with open('save/clf.pickle', 'rb') as f:
   clf2 = pickle.load(f)
   print(clf2.predict(X[0:1]))

# method 2: joblib
import joblib
# Save
joblib.dump(clf, 'save/clf.pkl')
# restore
clf3 = joblib.load('save/clf.pkl')
print(clf3.predict(X[0:1]))