当前位置：网站首页>Invalid classes inferred from unique values of `y`. Expected: [0 1 2], got [1 2 3]

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

2022-08-01 03:36:00 【川川菜鸟】

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

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

model.fit(X_train, y_train)

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

版权声明
本文为[川川菜鸟]所创，转载请带上原文链接，感谢
https://chuanchuan.blog.csdn.net/article/details/126091747