使用XGB训练中,出现处错误:Invalid classes inferred from unique values of `y`. Expected: [0 1], got ['0.0' '1.0']

原因:

发生这种情况是因为类列必须从 0 开始(自 1.3.2 版以来要求)。解决这个问题的一种简单方法是使用sklearn.preprocssing 库中的LabelEncoder

解决:

from sklearn.preprocessing import LabelEncoder
le = LabelEncoder()
y_train = le.fit_transform(y_train)

将y数据进行转换之后,就可以正常运行

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐