ValueError: невозможно преобразовать массив размером 3800 в форму (3800,224,224,3)

Question

мой код:

A = np.concatenate((folder[0], folder[1], folder[2], folder[3], folder[4], folder[5],
folder[6], folder[7], folder[8], folder[9], folder[10], folder[11]))

label = np.ones(4750,dtype = int)
label[0:263] = 0
label[263:653] = 1
label[653:940] = 2
label[940:1551] = 3
label[1551:1772] = 4
label[1772:2247] = 5
label[2247:2901] = 6
label[2901:3122] = 7
label[3122:3638] = 8
label[3638:3869] = 9
label[3869:4365] = 10
label[4365:4750] = 11

data,Label = shuffle(A,label,random_state=0)
train_data = [data, Label]
X,Y = [train_data[0],train_data[1]]

X_train, X_test, Y_train, Y_test = train_test_split(X, Y, test_size=0.2, random_state=1)

x_test, x_val, y_test, y_val = train_test_split(X_test, Y_test, test_size=0.2, random_state=1)



print(X_train.shape)

вывод: (3800, 268203)

print(f'train size: {X_train.shape}, val size: {x_val.shape}, test size: {X_test.shape}')

вывод: размер поезда: (3800, 268203), размер val: (760, 299, 299, 3), размер теста: (950, 268203)

X_train = X_train.reshape((X_train.shape[0],299, 299, 3))

x_val = x_val.reshape((x_val.shape[0],299, 299, 3))

x_test = x_test.reshape((x_test.shape[0],299, 299, 3))

, когда я использую np.zero вместо X_train.reshape, он работает, но выдает ошибку после построения модели. теперь я получаю ошибку:

   ValueError                                Traceback (most recent call last)
<ipython-input-79-bbe4db2d4820> in <module>
      3 X_val = X_val.reshape((X_val.shape[0],299,299,3))
      4 
----> 5 X_test = X_test.reshape((x_test.shape[0],299,299,3))

ValueError: cannot reshape array of size 254792850 into shape (760,299,299,3)

, пожалуйста, помогите мне решить эту проблему.