Exporting histograms from TensorFlow

histogram_blog

Lately I have been working a lot with Google’s TensorFlow library for machine learning. It has a really nice tool for data visualisation, TensorBoard, which can be very useful to understand how the training and evaluation of your model is working. One small bottleneck though is that it has a built-in tool for data export that only works for the scalar functions and unfortunately not for more complex visualisation means like histograms. I find especially the histograms to be particularly useful because they show how is your probability distribution narrowing as a function of learning steps, so it is a really useful figure of merit for understanding the training/evaluation. This is also the reason why I thought it would be useful to export the histograms and customise them for example in Matlab. I would like to share here the code for exporting the histograms from the TensorFlow model. Hopefully you will find it useful! You can download it here.


 

Screen Shot 2017-09-27 at 00.00.10.png

Leave a Reply

Fill in your details below or click an icon to log in:

WordPress.com Logo

You are commenting using your WordPress.com account. Log Out /  Change )

Google+ photo

You are commenting using your Google+ account. Log Out /  Change )

Twitter picture

You are commenting using your Twitter account. Log Out /  Change )

Facebook photo

You are commenting using your Facebook account. Log Out /  Change )

Connecting to %s