我的神经网络预测给我一个错误: IndexError:列表索引超出范围

import numpy as np

# 假设你的输入数据是 X
X = np.random.rand(100, 20)  # 100个样本，每个样本20个特征

# 确保 X 的形状正确
assert X.shape == (100, 20), "输入数据的形状不正确"

from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Dense

model = Sequential()
model.add(Dense(64, input_dim=20, activation='relu'))  # 输入层20个神经元，隐藏层64个神经元
model.add(Dense(10, activation='softmax'))  # 输出层10个神经元

from sklearn.model_selection import train_test_split

# 分割训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)

# 确保分割后的数据形状正确
assert X_train.shape[0] == y_train.shape[0], "训练集数据形状不匹配"
assert X_test.shape[0] == y_test.shape[0], "测试集数据形状不匹配"