def METHOD_NAME(predictions, classes):
    def prediction_to_proba(y_pred):
        proba = np.zeros(len(classes))
        proba[classes.index(y_pred)] = 1
        return proba
    return np.apply_along_axis(prediction_to_proba, axis=1, arr=predictions.reshape(-1, 1))