Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Tensorboard logging non-tensor (numpy) information (AUC)

I would like to record in tensorboard some per-run information calculated by some python-blackbox function.

Specifically, I'm envisioning using sklearn.metrics.auc after having run sess.run().

If "auc" was actually a tensor node, life would be simple. However, the setup is more like:

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

If there is a more tensorflow-onic way of doing this I am interested in that. My current setup involves creating separate train&test graphs.

If there is a way to complete the task as stated above, I am interested in that as well.

like image 233
user3391229 Avatar asked Jun 27 '16 17:06

user3391229


1 Answers

You can make a custom summary with your own data using this code:

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

Then you can add that summary to the summary writer yourself. (Don't forget to add a step).

like image 162
dandelion Avatar answered Oct 31 '22 15:10

dandelion