sklearn的模型如何保存下来?

  统计/机器学习 Python I/O    浏览次数:11068        分享
4

sklearn训练好的模型,有什么方法可以以某种文件的形式保存下来?以后用的时候就不需要在train fit了,直接读取模型,然后predict。


 

PG Two   2017-09-27 20:25



   2个回答 
12


#保存

from sklearn.externals import joblib

joblib.dump(model, 'path/to/my_model.pkl')

#读取

model = joblib.load('path/to/my_model.pkl') 


参阅: Model persistence

SofaSofa数据科学社区DS面试题库 DS面经

s3040608090   2017-09-28 09:03

5

也可以直接用pickle将模型保存到本地

import pickle
with open('filename.pkl', 'wb') as f:
    pickle.dump(clf, f)

读取

with open('filename.pkl', 'rb') as f:
    clf = pickle.load(f)


SofaSofa数据科学社区DS面试题库 DS面经

sasa   2017-10-12 05:27



  相关讨论

python如何读取元素中含有逗号的csv文件

怎么把statsmodels训练的模型本地保存?

怎么在python中读取mat格式数据文件?

python怎么读取txt格式的数据文件?

怎么把python的dict存为json格式文件

python能不能直接读取word文本,doc或者docx文件?

python 读json 错误 ValueError: Unexpected character found when decoding array value (2)

关于如何读取文件效率最高,最节约内存?

怎么把多个csv表格读进python然后合并成一个表格输出为csv?

把XGBRegressor保存到本地文件并调用

  随便看看

python里怎么求一个矩阵的秩?

统计里的IQR是什么意思?

协方差矩阵一定是半正定的吗?

概率论中的鞅是什么?

Random Forest可以用来做聚类?