首页 > 解决方案 > XGBoost:提供分类类型时,DMatrix 参数 `enable_categorical` 必须设置为 `True`

问题描述

我有一个数据框data,其列完全由 NUMERICAL 变量(整数和浮点数)和一个熊猫系列组成labels,其元素是与. 我想训练一个 XGBoost 分类器,但是当我使用调用时:"Exercise""Restdata

dtrain = xgb.DMatrix(data, label=labels)

enable_categorical我收到错误“当提供分类类型时,必须将DMatrix 参数设置为True”,然后错误消息打印出所有列名的列表data(我重复一遍,它只包含数字变量)。对我来说,列名似乎被解释为分类变量。我怎样才能解决这个问题?谢谢 :)

标签: pythonxgboost

解决方案


推荐阅读