0

我有一次性编码的标签。我想用它们来训练和预测一个 catboost 分类器。但是,当我拟合时它给了我一个错误,说标签的每行不允许多个整数值。那么 catboost 是否不允许标签的 one-hot 编码?如果没有,我怎样才能让 catboost 工作?

4

2 回答 2

0

catboost 在内部自动进行因子编码,无需手动进行

于 2020-11-05T09:13:20.430 回答
0

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

解决方法是将 one-hot 编码转换为分类值。当然,大多数时候,我们采用分类值并转换为 one-hot 编码。所以不要做这一步。

然后,将损失函数设置为“MultiClass”。这是 catboost(我认为大多数梯度提升包)将支持多分类的唯一损失函数。

于 2017-08-11T23:40:48.557 回答