使用加法而不是平均值的Keras Tensorflow聚合度量

2024-10-02 02:35:05 发布

您现在位置:Python中文网/ 问答频道 /正文

我知道Tensorflow/Keras提供了状态度量,可以使用metric.update_state()更新这些度量。我理解以有状态的方式更新度量是通过使用MeanMetricWrapper类获取平均值/平均值来执行的

如果我想使用另一个操作(例如,加法)来更新度量,我应该怎么做(假设我想在所有批次中累积损失,而不是取平均值,这样我可以在整个历元中打印损失,而不是每批次平均值)

我更感兴趣的是能够与model.train_on_batch()无缝协作的解决方案。多谢各位


Tags: model度量状态tensorflow方式updatetrainmetric

热门问题