У меня есть пример данных с 6 столбцами и 100 строками (все значения целые). Есть 20 классов, в которые входные данные классифицированы. Это модель, которую я пытался построить:
model = Sequential()
model.add(Dense(50,input_shape=X.shape[1:],activation='relu'))
model.add(Dense(20,activation='softmax'))
model.compile(loss='categorical_crossentropy', optimizer='rmsprop',
metrics=['accuracy'])
model.summary()
model.fit(X, Y, epochs=1000, verbose=0)
predictions=model.predict(test_data)
Однако я получаю сообщение об ошибке:
Error when checking target: expected dense_2 to have shape (20,) but got array with shape (1,)
У меня два вопроса:
- Что я делаю не так?
- Можете ли вы дать мне правильную архитектуру для этого?