问答交流

xgboost的模型如何保存和读取?

由xuguoyu创建,最终由xuguoyu 被浏览 60 用户

问题

问题描述

xgboost的模型如何保存和读取?

问题代码

m5 = M.xgboost.v1(
training_ds=mSR3data,
features=mSR5.data,
predict_ds=mSR4data,
num_boost_round=NUMROUND,
objective='排序(pairwise)',#其他如map,ndcg
booster='gbtree',
max_depth=MAXDEPTH,
key_cols='date,instrument',
group_col='date',
other_train_parameters={}
)
modellist.append(m5.output_model.id)

如何保存已经训练好的xgboost模型和读取保存的模型进行预测?

如果预测model=???

    m5 = M.xgboost.v1(
    #training_ds=mSR3data,
    features=mSR5.data,
    model=???,

    predict_ds=mSR4data,
    num_boost_round=NUMROUND,
    objective='排序(pairwise)',#其他如map,ndcg
    booster='gbtree',
    max_depth=MAXDEPTH,
    key_cols='date,instrument',
    group_col='date',
    other_train_parameters={}
    )

\

标签

XGBoostxgboost
评论
  • 我的意思是如果用DataSource 通过m5.output_model.id缓存,好像时间一长,训练的次数多了以后,原先保存的output_model.id就失效了。。。,有没有比较好的办法固化模型。 \ ``` m5.output_model.id,model=DataSource(m5.output_model.id),这个很容易失效 ``` \
  • 看看这里的文档有没有帮助呢?
  • ok,搞定
{link}