ValueError: Input 0 is incompatible with layer lstm_13: expected ndim=3, found ndim=4

前端 未结 4 1275
甜味超标
甜味超标 2020-12-14 17:40

I am trying for multi-class classification and here are the details of my training input and output:

train_input.shape= (1, 95000, 360) (95000 length

4条回答
  •  甜味超标
    2020-12-14 18:23

    I solved the problem by making

    input size: (95000,360,1) and output size: (95000,22)

    and changed the input shape to (360,1) in the code where model is defined:

    model = Sequential()
    model.add(LSTM(22, input_shape=(360,1)))
    model.add(Dense(22, activation='softmax'))
    model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=['accuracy'])
    print(model.summary())
    model.fit(ml2_train_input, ml2_train_output_enc, epochs=2, batch_size=500)
    

提交回复
热议问题