Tensorboard记录非张量(numpy)信息(AUC)

Question

我想在张量板中记录一些python-blackbox函数计算出的每次运行信息.

具体来说,我正在设想在运行sess.run()之后使用sklearn.metrics.auc.

如果"auc"实际上是一个张量节点,那么生活就会很简单.但是,设置更像是:

stuff=sess.run()
auc=auc(stuff)

如果有更多的tensorflow-onic方式这样做我感兴趣.我目前的设置涉及创建单独的列车和测试图表.

如果有办法完成上述任务,我也对此感兴趣.

Answer 1

您可以使用以下代码使用自己的数据制作自定义摘要:

tf.Summary(value=[tf.Summary.Value(tag="auc", simple_value=auc)]))

然后,您可以自己将摘要添加到摘要编写器.(别忘了添加一个step).