您的位置:首页 > 其它

sklearn模型的保存与恢复

2017-05-26 22:29 281 查看
举例说明:

模型的保存过程:

from sklearn.externals import joblib
from sklearn import svm
import os

# os.chdir("workspace/model_save")
X = [[0, 0], [1, 1]]
y = [0, 1]
clf = svm.SVC()
clf.fit(X, y)
joblib.dump(clf, "train_model.m")模型的读取过程:
from sklearn.externals import joblib

clf = joblib.load("train_model.m")
X = [[0, 0], [1, 1]]
y = [0, 1]

print(clf.predict(X))
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  机器学习