如何解决如何将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 举报,一经查实,本站将立刻删除。