You're almost done: Please save `predicted_labels` in the same (non-binary) format like the training data.  This allows us to view and manipulate the labels in arbitrary ways, because viewing/manipulating textual data is so easy on UNIX.