code: def create_proba_result(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)
      )
  
imports: import numpy
label_desc: Use image node_id 1 for calling a global function with example usage: create_proba_result(predictions, classes) and returns: np, proba