weixin_43931696的博客

原创 当样本分布不均衡的时候如何修改spark 版本的xgboost的损失函数

spark xgboost自定义损失函数和评价函数一、spark xgboost自定义损失函数xgboost定义损失函数的是通过setCustomObj定义，要想定义损失函数，首先我们得知道其接收的数据类型是什么类型的可见它需要的数据类型是ObjectiveTrait的实现类。可见，我们只需重载getGradient方法即可。代码如下/** * 自定义损失函数 */...