2017-08-11 123 views
0

我有一个热点编码的标签。我想用它们来训练和预测一个catboost分类器。然而,当我合适时,它给我一个错误,说标签每行不允许有多个整数值。那么catboost不允许对标签进行单热编码?如果没有,我怎样才能让catboost工作?Catboost是否支持单热编码?

+0

你传递给'catboost'的参数是什么?你有没有加入'one_hot_max_size'? –

+0

@BobDalgleish是的......我只是通过了learning_rate,depth,iterations和one_hot_max_size ....我找到了解决方法,尽管...... – TanMath

回答

0

我找到了解决此问题的解决方法。这个问题可能有更好的解决方案,我很乐意听到。

解决方法是将单热编码转换为分类值。当然,大多数情况下,我们将我们的分类值转换为单热编码。所以只是不要做这一步。

然后,设置损失函数为'MultiClass'。这是catboost(我认为大多数渐变增强软件包)将支持多分类的唯一损失函数。