Как нарисовать несколько кривых PR на одном графике для PyTorch TensorBoard? ⇐ Python

[Anonymous]

Ниже приведен код, рисующий кривые PR, взятые из https://pytorch.org/tutorials/intermedi ... ensorboard

Код: Выделить всё

from torch.utils.tensorboard import SummaryWriter

writer = SummaryWriter()

def add_pr_curve_tensorboard(class_index, test_probs, test_preds):
tensorboard_preds = test_preds == class_index
tensorboard_probs = test_probs[:, class_index]

writer.add_pr_curve(classes[class_index],
tensorboard_preds,
tensorboard_probs,
global_step=0)
writer.close()

for i in range(len(classes)):
add_pr_curve_tensorboard(i, test_probs, test_preds)

Я подумал, что если я изменю тег первого параметра на то же значение «all_pr_curves», это поможет нарисовать все кривые PR на одном графике.

Код: Выделить всё

writer.add_pr_curve(
'all_pr_curves',
tensorboard_preds,
tensorboard_probs,
global_step=0)

Но оказывается, что более поздний перезапишет предыдущий. Таким образом, в конце отображается только последняя кривая PR.
Есть ли способ нарисовать все кривые PR на одном графике?

Подробнее здесь: https://stackoverflow.com/questions/648 ... ensorboard