天天看點

Demo4 - 加載模型

當已經訓練好的模型如何直接使用,。

參考demo3, 兩種load方式,根據你選擇的是pickle,還是joblib方式。

代碼:

# -- coding: utf-8 --
# 通過本個demo 将學習後的模型的加載,無需重新訓練模型。
# (模型的保留的遷移- 參考demo3)

# 方法1,使用python自帶的pickle
import pickle
from sklearn import datasets

# 導入模型方法1:
# file = open('saved_model/rfc.pickle','rb')
# rfc = pickle.load(file)  # !!!!重點
# file.close()

#導入模型方法2:
from sklearn.externals import joblib
rfc = joblib.load('saved_model/rfc.pkl')

# 導入後驗證
(X,y) = datasets.load_iris(return_X_y=True)  # 導入測試資料
print('驗證結果是:')
print(rfc.predict(X[0:1,:]))