@staticmethod
def _map_preds(pred):
    """Map the predicted label to another integer or char to be written
        to .csv file"""
    if pred == 0:
        res = 'w'
    elif pred == 1:
        res = 'n'
    elif pred == 2:
        res = 'r'
    else:
        res = pred.cpu().numpy() - 2
    return str(res)
