有没有方法可以在MLR框架中为XgBoost设置“基本边距”？

Question

我正在尝试在MLR框架中适应XgBoost模型。尽管该框架有充分的文档记录，但XgBoost库的某些细节我无法在MLR中复制，尤其是“基本余量”。在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中未实现。我们没有任何计划来支持它，但我们随时欢迎您提供pull request。