确定张量流keras LSTM的输入形状?

时间:2018-07-26 20:24:04

标签: tensorflow recurrent-neural-network

我对此有点麻烦。首先,这是我的数据:

test_data,test_labels,train_data,train_labels

train_data[0]

[1, 5, 5, 0, 0, 1, 1, 1, 25, 1, 1, 10, 0, 1, 1, 1, 0, 1, 39, 2, 0, 1, 1, 12, 3]

train_labels[0]

0

test_data和test_labels完全相同(只是输入数据的50/50分割)。 test_data中每个数组的数组大小将始终为25个元素。标签为0表示好或1表示差。

现在,到目前为止,我已经尝试了很多方法,但还没有想出如何重塑这些数组的形状。我实质上是想这样做:

    model.add(keras.layers.LSTM(256, input_shape=unknown, return_sequences=False, return_state=False, dropout=0.2))
    model.add(keras.layers.Dense(256))
    model.add(keras.layers.Dropout(0.3))
    model.add(keras.layers.Dense(2, activation=tf.nn.softmax))
    history = self.model.fit(self.train_data,
                             self.train_labels,
                             epochs=50,
                             batch_size=64,
                             verbose=1,
                             validation_split=0.2)

另一个问题,最后一个密集层2是否正确,或者在这种情况下应该为1?

0 个答案:

没有答案