我对此有点麻烦。首先,这是我的数据:
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?