ValueError: Найдены входные переменные с непоследовательным количеством выборок: [10000, 5711]

Question

Я пытаюсь обучить нейронную сеть с помощью функции BFGS в качестве функции обновления, при обучении и получении функции стоимости, похоже, не возникает никаких проблем, однако, когда я пытаюсь проверить ее и сделать прогноз, он выдаст мне ошибку:

ValueError: Найдены входные переменные с непоследовательным количеством выборок: [10000, 5711] код, который я использую:

X=data.iloc[:,0:].values
y=data.iloc[:,-1:].values

X=pd.DataFrame(data,columns=data.columns)
y=pd.DataFrame(data,columns=['class'])

X.drop(['class'],axis=1,inplace=True)
X_train,X_test,y_train,y_test=train_test_split(X,y,test_size=0.3,random_state=0)

sc=StandardScaler()
X_train=sc.fit_transform(X_train)
X_test=sc.transform(X_test)

qnewton=algorithms.QuasiNewton(
network=[
    layers.Input(13),
    layers.Tanh(10),
    layers.Sigmoid(1),
],
update_function='bfgs',verbose=True
)
qnewton.train(X_train,y_train)
y_predict=qnewton.predict(X_test)

Я не совсем понимаю, что я делаю неправильно. Пожалуйста, помогите.