欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Sklearn官方文档:保存模型(Model persistence)

程序员文章站 2022-06-12 22:45:46
...
#sklearn official doc.
#Save model:
#1. piclke
>>> import pickle
>>> s = pickle.dumps(clf)
>>> pickle.loads(s)

#2. joblib
>>> from joblib import dump, load
>>> dump(clf, 'filename.joblib') 
>>> clf = load('filename.joblib') 

对于处理大的数据,joblib往往更有效,但joblib只能够存储为文件,而不能储存在缓存中。


另外一种方法是采用JSON:

JSON具有比pickle更好的可读性,

JSON is a text serialization format (it outputs unicode text, although most of the time it is then encoded to utf-8), while pickle is a binary serialization format;

JSON可用于更广泛的其他编程语言而pickle只能用于python;

在python中,pickle更加强大,支持的python类型更多。

相关标签: sklearn