r语言 - 有没有办法在 MLR 框架中为 XgBoost 设置"base margin"？

Question

我正在尝试在MLR框架中安装XgBoost模型。虽然该框架有相当好的文档，但我无法在 MLR 中复制 XgBoost 库的一些细节，特别是"基本边距"。 在 XgBoost 库中，我只需设置：

db_xgbmatrix <- xgb.DMatrix(db)
setinfo(db_xgbmatrix , "base_margin", margin)

然后我可以继续训练模型;而在 MLR 中，一旦我创建了任务和学习者：

tsk      <- makeRegrTask(data =  db, target = target_var)
lrn <- makeLearner("regr.xgboost", predict.type = "response", eta = 0.1,
max_depth = 8, min_child_weight = 20,
subsample = 0.75, colsample_bytree = 0.75,
nrounds = 100, nthread = cl_n, objective="count:poisson")

我不太确定应该在哪里设置基本边距。有什么想法吗？该功能是否实现并隐藏在某个地方？ 提前谢谢大家

Answer 1

这在 mlr 中没有实现。我们没有任何计划来支持它，但随时欢迎您贡献拉取请求。