How to get the real prediction?

[celsofranssa]

The predicted labels' ids for a given text are mapped to other ids, as below:

        true_labels = set([label_map[i] for i in true_labels.split()])

How could someone save the predictions for each test text using the raw labels' ids assigned to each text?