Pytorch中使用tensorboard添加pr曲线的方法

这篇文章主要介绍“Pytorch中使用tensorboard添加pr曲线的方法”，在日常操作中，相信很多人在Pytorch中使用tensorboard添加pr曲线的方法问题上存在疑惑，小编查阅了各式资料，整理出简单好用的操作方法，希望对大家解答”Pytorch中使用tensorboard添加pr曲线的方法”的疑惑有所帮助！接下来，请跟着小编一起来学习吧！

代码:

from torch.utils.tensorboard import SummaryWriterimport numpy as np
np.random.seed(20200910)labels = np.random.randint(2, size=100)  # binary labelpredictions = np.random.rand(100)writer = SummaryWriter()writer.add_pr_curve('pr_curve林麻子', labels, predictions, 0)writer.close()# tensorboard --logdir=runs

运行结果展示:

到此，关于“Pytorch中使用tensorboard添加pr曲线的方法”的学习就结束了，希望能够解决大家的疑惑。理论与实践的搭配能更好的帮助大家学习，快去试试吧！若想继续学习更多相关知识，请继续关注亿速云网站，小编会继续努力为大家带来更多实用的文章！