Confusion matrix fix
What does this MR do?
- in the Tau MR the ordering of the class names for the confusion matrix was changed which results in wrong plots with our default plotting configs.
- Some small fixes for the umami training
- Some minor convenience stuff