- 博客(1)
- 收藏
- 关注
原创 当样本分布不均衡的时候如何修改spark 版本的xgboost的损失函数
spark xgboost自定义损失函数和评价函数一、spark xgboost自定义损失函数xgboost定义损失函数的是通过setCustomObj定义,要想定义损失函数,首先我们得知道其接收的数据类型是什么类型的可见它需要的数据类型是ObjectiveTrait的实现类。可见,我们只需重载getGradient方法即可。代码如下/** * 自定义损失函数 */...
2019-11-19 20:50:47
803
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人