tensorflow - 如何在训练期间在张量板上显示每个班级的 f1 分数

Question

我正在训练一个多类分类网络。我使用来自 tensorboard-addons 的指标并像这样编译模型：

import tensorflow-addons as tfa

[...]

metrics = [tfa.metrics.F1Score(num_classes=CLASSES, average=None)]
model.compile(optimizer='adam', loss=tf.keras.losses.CategoricalCrossentropy(), metrics=metrics)

平均无应给出每个班级的 f1 分数。如果我用

result = model.evaluate(dataset, return_dict=True)

正如预期的那样，我得到了一个包含每个班级的 f1 分数的数组。
然而，在训练期间，我在张量板上只得到一个分数和一个图表。如何在张量板训练期间显示每个时期的班级明智 f1 分数。

我正在使用 tensorflow 2.5 。