Проблема в вашей X_train, y_train, X_test, y_test = train_test_split (X, y, test_size = 0.2) части.
Согласно документации scikit-learn в здесь , правильный порядок возвращаемого значения функции train_test_split :
- X_train,
- X_test,
- y_train,
- y_test
Ваш заказ в коде неправильный.Давайте заменим строку, которую вы использовали ** train_test_split **, на эту строку:
X_train, X_test, y_train, y_test = train_test_split(X,y, test_size=0.2)
Надеюсь, это решит вашу проблему.