时间戳

首页 > 解决方案 > 是否有任何“官方”方法可以将 scikit-learn 模型保存到/从纯文本（如 json）中保存/加载？

python - 是否有任何“官方”方法可以将 scikit-learn 模型保存到/从纯文本（如 json）中保存/加载？

问题描述

我想将任何 sklearn 模型保存到 json，并想要一个 scikit-learn 的官方方式。你有什么想法？

我需要这种代码：

from sklearn.export import json_export
from sklearn.ensemble import RandomForestClassifier

clf = RandomForestClassifier().fit(X, y)

with open("model.json", "w") as f:
    json_export(clf, f)

标签： pythonscikit-learn

解决方案

不幸的是，sklearn 中没有官方的方法。使用 pickle/joblib 序列化分类器 Python 对象，或为您的特定模型实现您自己的 JSON 序列化。按照惯例，由训练设置的属性以 _ 结尾，但您还需要许多构造函数参数。

推荐阅读