亚洲在线久爱草,狠狠天天香蕉网,天天搞日日干久草,伊人亚洲日本欧美

為了賬號安全,請及時綁定郵箱和手機立即綁定

機器學習模型的保存與加載——基于pickle模塊和Scikit-learn的joblib

標簽:
機器學習

1.训练好一个Model以后需要保存和再次预测
2.有两个模块用来保存模型 : pickle和joblib
3.Sklearn的模型导出本质上是利用Python的Pickle机制。对Python的函数进行序列化,也就是把训练好的Transformer函数序列化并存为文件。

代码流程:
1.保存Model(注:save文件夹要预先建立,否则会报错)

joblib.dump(clf, ‘save/clf.pkl’)

2.读取Model

clf2 = joblib.load(‘save/clf.pkl’)

3.测试读取后的Model

print(clf2.predict(X[0:1]))**

实例一,SVM分类模型

import pickle 
from sklearn.externals import joblib
from sklearn.svm import SVC
from sklearn import datasets

#定义一个分类器
svm = SVC()

iris = datasets.load_iris()
X = iris.data
y = iris.target

#训练模型
svm.fit(X,y)

#1.保存成Python支持的文件格式Pickle
#在当前目录下可以看到svm.pickle
with open('svm.pickle','wb') as fw:
	pickle.dump(svm,fw)
#加载svm.pickle
with open('svm.pickle','rb') as fr:
	new_svm1 = pickle.load(fr)

# print (new_svm1.predict(X[0:1]))

#2.保存成sklearn自带的文件格式Joblib
joblib.dump(svm,'svm.pkl')
#加载svm.pkl
new_svm2 = joblib.load('svm.pkl')
print (new_svm2.predict(X[0:1]))

运行结果
图片描述
图片描述

————————————————————
补充:
图片描述
——————————————————————

实例二,随机森林模型

#随机森林
from sklearn.ensemble  import RandomForestClassifier
from sklearn.datasets  import load_iris
from sklearn.model_selection import train_test_split
from sklearn.metrics   import classification_report 
from sklearn.externals import joblib

#1.保存模型
data = load_iris()
X,y = data["data"],data["target"]
X_train,X_test,y_train,y_test = train_test_split(X,y,test_size = 0.33,random_state = 42)
clf = RandomForestClassifier(max_depth = 2,random_state = 0)
clf.fit(X_train,y_train)

#print (clf.feature_importances_)
"""
print (
	classification_report(y_test,clf.predict(X_test),target_names = data["target_names"])
	)
"""
joblib.dump(clf,"classification.pkl")


#2.加载模型
clf2 = joblib.load('classification.pkl')
print (clf2.feature_importances_)
print (
	classification_report(y,clf.predict(X),target_names = data["target_names"])
	)

运行结果1
图片描述
运行结果2
图片描述

點擊查看更多內容
1人點贊

若覺得本文不錯,就分享一下吧!

評論

作者其他優質文章

正在加載中
算法工程師
手記
粉絲
23
獲贊與收藏
45

關注作者,訂閱最新文章

閱讀免費教程

感謝您的支持,我會繼續努力的~
掃碼打賞,你說多少就多少
贊賞金額會直接到老師賬戶
支付方式
打開微信掃一掃,即可進行掃碼打賞哦
今天注冊有機會得

100積分直接送

付費專欄免費學

大額優惠券免費領

立即參與 放棄機會
微信客服

購課補貼
聯系客服咨詢優惠詳情

幫助反饋 APP下載

慕課網APP
您的移動學習伙伴

公眾號

掃描二維碼
關注慕課網微信公眾號

舉報

0/150
提交
取消