保存模型的 兩種方法

from sklearn.linear_model import LogisticRegression
from sklearn import datasets
import pickle
from sklearn.externals import joblib

clf = LogisticRegression()
iris = datasets.load_iris()
X, y = iris.data, iris.target
clf.fit(X, y)
print(clf.score(X, y))

with open('clf.pickle', 'wb') as f:
    pickle.dump(clf, f)
    
joblib.dump(clf, 'clf.pkl')
del clf

以上是以兩種方法將模型保存, 以便以後直接讀取模型進行預測

可見這兩種讀取模型的方法都是可以的, 效果相同