H2O 中的自定义损失函数

Question

我正在通过 R 使用 H2O。我正在尝试构建随机森林、XGBoost、GBM 模型来解决多类问题。

H2O 提供的模型性能洞察力很棒，但作为成功标准之一，我有自己的自定义函数，当模型用于对一组用户（例如验证集）进行评分时，该函数会对模型的准确性进行评分。一般来说，这个函数奖励预测并惩罚错误预测。鉴于此问题中的目标类别是序数类别，惩罚分数取决于错过预测的位置数。这个自定义函数的反函数也可以被认为是一个损失函数。

我想知道在训练模型时是否以及如何将这种自定义损失函数插入 H2O。

Answer 1

虽然目前不支持自定义损失函数（尽管有一个jira 票可以添加它），但对自定义评估函数有一些支持，您可以在此处找到更多详细信息：https : //github.com/ h2oai/h2o-3/blob/master/h2o-docs/src/dev/custom_functions.md。