是否有任何“官方”方法可以将scikit学习模型保存到json之类的纯文本/从中加载？

Question

我想将任何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)

Answer 1

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