机器学习(六)使用sklearn库的model存储


# -*- coding: utf-8 -*-
"""
Created on Tue Oct 25 21:39:04 2016

@author: Administrator
model说明,normalization
"""

from sklearn import datasets
from sklearn.cross_validation import train_test_split
from sklearn.linear_model import LinearRegression


load_data=datasets.load_boston()
data_x=load_data.data
data_y=load_data.target

x_train,x_test,y_train,y_test=train_test_split(data_x,data_y,test_size=0.1)

model=LinearRegression()
model.fit(x_train,y_train)

"""
两种存储model的形式

"""

"""
method 1: pickle
"""
import pickle

#save
with open('clf.pickle','wb') as f:
    pickle.dump(model,f)

#restore
with open('clf.pickle','rb') as f:
    model2=pickle.load(f)
print model2.predict(data_x[:10,:])
print data_y[:10]

print ' +----------------------------------------+'
"""
method 2: sklearn 外部库joblib
"""
from sklearn.externals import joblib
#save
joblib.dump(model,'clf2.pkl')
#restore
model3=joblib.load('clf2.pkl')
print model3.predict(data_x[10:20,:])
print data_y[10:20]
  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值