【发布时间】:2016-09-29 07:33:01
【问题描述】:
我正在使用一个大数据集,因此我正在尝试使用 train_on_batch(或适合 epoch = 1)
model = Sequential()
model.add(LSTM(size,input_shape=input_shape,return_sequences=False))
model.add(Dense(output_dim))
model.add(Activation('softmax'))
model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=["accuracy"])
for e in range(nb_epoch):
for batch_X, batch_y in batches:
model.train_on_batch(batch_X,batch_y)
# or
# model.fit(batch_X,batch_y,batch_size=batch_size,nb_epoch=1,verbose=1,shuffle=True,)
但是当训练开始时,会发生这种情况:
(0, 128)
Epoch 1/1
128/128 [==============================] - 2s - loss: 0.3262 - acc: 0.1130
(129, 257)
Epoch 1/1
128/128 [==============================] - 2s - loss: -0.0000e+00 - acc: 0.0000e+00
无论我等待多少个 epoch,它都不会改变。即使我更改批量大小,也会发生同样的事情:第一批具有良好的值,然后再次变为“loss: -0.0000e+00 - acc: 0.0000e+00”。
有人可以帮助了解这里发生了什么吗?
【问题讨论】:
-
如果您的训练数据包含非常少量的独特示例并且您的网络在第一批中学习了所有这些示例,则可能会发生这种情况。也许您不小心在数据集创建脚本中使用数组引用而不是副本来放置相同的元素。
-
是的,看看预测和标签,看看网络是否真的达到了 0 准确度。这将帮助您调试。
-
@DmitryKostyaev 相同的元素。这是一个小错误,我觉得很傻。感谢您的帮助。
-
投票结束:(1) Keras 自 4 年前以来发生了巨大变化; (2) 没有足够的调试细节; (3) 这是 OP 的唯一问题,因此 (2) 不太可能得到解决。
标签: python batch-processing training-data keras lstm