r语言 - H2O 中的自定义损失函数



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

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

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

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

相关内容

  • 没有找到相关文章

最新更新