如何将xgboost模型另存为json的json？

问题描述

训练完xgboost模型后，我想将其和其他一些自定义字段另存为json对象，如下所示。目的是让我以后可以加载json对象，使用模型对象进行预测以及检查其他自定义字段。

model = xgb.train(params=tree_params,dtrain=data)
my_model_dict = {
    "model": ...<json serializable model object>...,# need help here
    "features": model.feature_names,"tree_params": tree_params,...
}
json.dumps(my_model_dict,file_path)

my_model_dict = json.load(file_path)
model = my_model_dict["model"]
predictions = model.predict(new_data)

是否可以将xgboost模型对象转换为json可序列化的对象，然后可以将其加载以进行标准的xgboost预测？

我很高兴可以使用

将原始模型另存为json。

model.save_model("my_model.json")
model = xgb.Booster("my_model.json")
model.predict(new_data)

但实际上我想做的是创建一个包含模型以及其他自定义字段的字典，该字典可以另存为json，然后加载以进行预测。

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）

json serialization xgboost