【发布时间】:2020-06-23 19:42:27
【问题描述】:
我正在使用 TensorFlow 训练 LSTM 模型。我们知道,在训练过程中,每个 epoch 的 loss 和 val_loss 报告是训练和测试数据集损失的平均值。我打算跟踪火车数据集中特定样本的丢失(特定日期)。另外,应该注意的是,我在 fit 函数中洗牌训练数据。
【问题讨论】:
-
您可以创建一个特定的操作来计算该样本的损失,并在您需要损失值或将其放在摘要中时调用该操作。在纯 TensorFlow 的情况下,它将在会话运行中完成,而在 fit 函数(keras 方式)的情况下,您必须编写自定义回调。
-
@talos1904 您能否将其添加为答案并提供更多信息?
-
下面添加的代码
标签: python tensorflow keras lstm epoch