跳到内容

此回调函数允许保存 xgb 模型文件,可以每隔 save_period 周期性保存,也可以在结束时保存。

不会在 booster 中留下任何属性。

用法

xgb.cb.save.model(save_period = 0, save_name = "xgboost.ubj")

参数

save_period

每经过 save_period 次迭代后将模型保存到磁盘;0 表示在结束时保存模型。

save_name

保存的模型文件的名称或路径。它可以包含一个 sprintf() 格式说明符,用于在文件名中包含整数迭代次数。例如,如果 save_name = 'xgboost_%04d.model',则在第 50 次迭代保存的文件将命名为 "xgboost_0050.model"。

一个 xgb.Callback 对象,可以传递给 xgb.train(),但不能传递给 xgb.cv()