It would be cool if the test script only writes the "predicted_labels" to the output file, so we can read it by external programs.

Secondly, if the evaluation code could be outsourced to an extra script ```evaluate_multiclass_labels.py```, we can re-use it for everything that looks like multiclass data.