Я новичок в этом, так что если кто-то может объяснить ошибку, это поможет мне много
код:
train_image, test_image, train_label, test_label = train_test_split(X, Y, test_size=0.30,random_state=42)#splits data, 30% for test and 70% for train
train_image = train_image.reshape(train_image.shape[0],32,32,3)
test_image = test_image.reshape(test_image.shape[0],32,32,3)
train_label = to_categorical(train_label, num_classes=len(data['Class']))
test_label = to_categorical(test_label, num_classes=len(data['Class']))
model = Sequential()
conv_01 = Conv2D(filters = 32,kernel_size=(3,3),activation='relu',input_shape=(32,32,3))
model.add(conv_01)
conv_02 = Conv2D(filters=64,kernel_size=(3,3),activation='relu')
model.add(conv_02)
pool = MaxPool2D(pool_size = (2,2),strides = (2,2), padding = 'same')
model.add(pool)
conv_11 = Conv2D(filters=64,kernel_size=(3,3),activation='relu')
model.add(conv_11)
pool_2 = MaxPool2D(pool_size=(2,2),strides=(2,2),padding='same')
model.add(pool_2)
drop = Dropout(0.3)
model.add(drop)
conv_out = Conv2D(filters = 3,kernel_size=(1,1),activation='softmax')
model.add(conv_out)
model.compile(optimizer='Adam',loss='categorical_crossentropy',metrics=['accuracy'])
model.fit(train_image,train_label,epochs=10,verbose = 1,validation_data=(test_image,test_label))
tes_loss,test_acc = model.evaluate(test_image,test_label)
prediction = model.predict(test_image)
train_image.shape: (66256, 32, 32, 3)
test_image.shape: (28396, 32, 32, 3)
train_label.shape: (66256, 58)
test_label.shape: (28396, 58)
Сообщение об ошибке:
Traceback (most recent call last):
File "processing.py", line 59, in <module>
model.fit(train_image,train_label,epochs=10,verbose = 1,validation_data=(test_image,test_label))
File "/home/mihir/Desktop/myenv/lib/python3.5/site-packages/tensorflow/python/keras/engine/training.py", line 1278, in fit
validation_split=validation_split)
File "/home/mihir/Desktop/myenv/lib/python3.5/site-packages/tensorflow/python/keras/engine/training.py", line 917, in _standardize_user_data
exception_prefix='target')
File "/home/mihir/Desktop/myenv/lib/python3.5/site-packages/tensorflow/python/keras/engine/training_utils.py", line 182, in standardize_input_data
'with shape ' + str(data_shape))
ValueError: Error when checking target: expected conv2d_3 to have 4 dimensions, but got array with shape (66256, 58)