Эта ошибка возникает из-за того, что ваши X и Y не имеют одинаковую длину (что требуется для train_test_split), то есть X.shape [0]! = Y.shape [0]. попробуйте это:
>>> X.shape
>>> Y.shape
И затем исправьте это:
- Вы должны удалить дополнительный список изнутри np.array () при определении X или удалить дополнительное измерение затем с помощью следующей команды: X = X.reshape (X.shape [1:]).
- И затем транспонируйте X, запустив X = X.transpose (), чтобы получить равное количество выборок в X и Y .