You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

使用HistGradientBoostingClassifier处理分类变量时触发ValueError的问题求助

HistGradientBoostingClassifier处理分类变量时触发ValueError的问题求助

我最终选择了HistGradientBoostingClassifier,因为它原生支持分类变量,不需要做独热编码或者标签编码。但当我尝试把分类变量传入模型并执行拟合时,却触发了如下错误:

ValueError: could not convert string to float: 'A'

下面是我复现问题的极简代码示例:

X = pd.DataFrame({
    'feature_1': [1, 2, 3, 1, 2],
    'feature_2': ['A', 'B', 'C', 'A', 'B']
})

y = pd.Series([0, 1, 0, 1, 0])  

X['feature_2'] = X['feature_2'].astype('category')

categorical_features = [1]  

model = HistGradientBoostingClassifier(categorical_features=categorical_features)

model.fit(X, y) 

我的分类变量并不是有序的,所以不太想用标签编码来处理。请问是不是我哪里操作遗漏了?

备注:内容来源于stack exchange,提问作者aaron_paynev4

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.04.15 03:19:36