Fix bugs in CTM calcs. Get perplexity code working for STM tests

For CTM, I forgot to reorder the results (tho re-ordering correctly
leads to _worse_ results than using incorrectly static input)

For STM there were multiple issues.
Add a mix of graphs types for STM

Two approaches, an all in one graph, or 12 individual graphs
Use one colorbar for the 6 imshow subplots
Previous commit apparently didn't include the change to the colorers

This was to have one colour bar for all six subplots. Fixed now
Fix bug with multiple plots of perplexities

Need to fix a bug with perplexity calculation
Use one subplot for the three configs of the CTM results

The three configs are the first fold, the first three folds, and the
first five folds
Print raw perplexity numbers after the graphs for the 5 folds case