Getting test prediction labels per image about capsnet-tensorflow

[naturomics]

I created a tool to visualize the predicted labels. you can use it directly.

The idea of that tool is get all the predicted labels(10 labels) for each sample, then we get an matrix with shape [num_sample, num_labels]. For each label, plot the point (sample_id, probability) in the R^2 coordinate. Finally you will get a cool picture, see the gif example here.

The tool can be found here.

What you need to do in the code is save the predicted labels/activation probability, and just plot it. See my CapsLayer repo for details.