将分类器保存到scikit-learn中的磁盘

将分类器保存到scikit-learn中的磁盘,第1张

将分类器保存到scikit-learn中的磁盘

分类器只是可以像其他任何东西一样被腌制和倾倒的对象。继续您的示例:

import cPickle# save the classifierwith open('my_dumped_classifier.pkl', 'wb') as fid:    cPickle.dump(gnb, fid)# load it againwith open('my_dumped_classifier.pkl', 'rb') as fid:    gnb_loaded = cPickle.load(fid)

编辑:如果您使用的是sklearn管道,其中有无法通过pickle(也不能通过joblib)序列化的自定义转换器,则使用Neuraxle的自定义ML管道保存是一种解决方案,您可以在每个管道上定义自己的自定义步骤保存程序步骤基础。如果在保存时定义了保存程序,则每个步骤都会调用保存程序,否则,对于没有保存程序的步骤,默认使用joblib。



欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/zaji/5673583.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-12-16
下一篇 2022-12-16

发表评论

登录后才能评论

评论列表(0条)

保存