跳到内容

此回调函数允许保存一个 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()