Invalid classes inferred from unique values of `y`. Expected: [0 1 2], got [1 2 3]

解决方法:对y_train进行编码再训练

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

model.fit(X_train, y_train)

成功:
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/weixin_46211269/article/details/126091747