H2O中的自定义损失功能

问题描述 投票:3回答:1

我是H2O的新手,我通过R访问它。我正在尝试构建随机森林,XGBoost,GBM模型来解决多类问题。 H2O提供的模型性能见解很棒,但作为成功标准之一,我有自己的自定义函数,当模型用于对一组用户进行评分时,可以对模型精度进行评分,即验证集。一般而言,该功能奖励预测并惩罚误预测。鉴于此问题中的目标类别是序数类别,惩罚分数取决于错过预测的位置数量。此自定义函数的反函数也可以被视为损失函数。

我想知道是否以及如何在培训模型时将这种自定义损失函数插入H2O。非常感谢这方面的任何帮助。

非常感谢

r machine-learning h2o
1个回答
4
投票

虽然目前不支持自定义丢失功能(虽然有一个jira ticket可以添加它),但是有一些支持自定义评估功能,你可以在这里找到更多细节:https://github.com/h2oai/h2o-3/blob/master/h2o-docs/src/dev/custom_functions.md

© www.soinside.com 2019 - 2024. All rights reserved.