У меня есть модель Keras со следующей сводной информацией, сохраненной в файл.
Layer (type) Output Shape Param #
=================================================================
conv2d_1 (Conv2D) (None, 70, 8, 64) 640
_________________________________________________________________
dense_1 (Dense) (None, 70, 8, 720) 46800
_________________________________________________________________
dense_2 (Dense) (None, 70, 8, 720) 519120
_________________________________________________________________
dense_3 (Dense) (None, 70, 8, 100) 72100
_________________________________________________________________
conv2d_2 (Conv2D) (None, 68, 6, 32) 28832
_________________________________________________________________
flatten_1 (Flatten) (None, 13056) 0
_________________________________________________________________
dense_4 (Dense) (None, 2) 26114
=================================================================
Total params: 693,606
Trainable params: 693,606
Non-trainable params: 0
_________________________________________________________________
None
Мои формы данных обучения и тестирования: (120,72,10,1) и (20,72,10, 1) соответственно. Когда я пытаюсь переобучиться, модель не принимает ничего меньшего, чем (4,72,10,1), когда я пытаюсь тренироваться на одной точке данных (1,72,10,1). Ниже приводится ошибка, которую я получаю.
Traceback (most recent call last):
File "./classifier.py", line 62, in <module>
classifier.fit(test_point, y_test[i])
File "/home/jazz/.local/lib/python3.5/site-packages/keras/engine/training.py", line 952, in fit
batch_size=batch_size)
File "/home/jazz/.local/lib/python3.5/site-packages/keras/engine/training.py", line 789, in _standardize_user_data
exception_prefix='target')
File "/home/jazz/.local/lib/python3.5/site-packages/keras/engine/training_utils.py", line 138, in standardize_input_data
str(data_shape))
ValueError: Error when checking target: expected dense_4 to have shape (2,) but got array with shape (1,)
Я не знаю, как двигаться дальше, любая помощь приветствуется.