使用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数据进行转换之后,就可以正常运行