参考链接: add_pr_curve
代码:
from torch.utils.tensorboard import SummaryWriter
import numpy as np
np.random.seed(20200910)
labels = np.random.randint(2, size=100) # binary label
predictions = np.random.rand(100)
writer = SummaryWriter()
writer.add_pr_curve('pr_curve林麻子', labels, predictions, 0)
writer.close()
# tensorboard --logdir=runs
运行结果展示: