code: def get_pred_col(preds, target):
      t = target.lower()
      for c in preds.keys():
          if c.lower() == t:
              if "probabilities" in preds[c]:
                  return preds[c]["probabilities"]
              else:
                  return preds[c]["predictions"]
      raise ValueError(
          f"Unable to find target column {t} in {preds.keys()}"
      )
  
file_path: ludwig/ludwig/explain/util.py
imports: from copy import deepcopy
  import pandas
  import torch
  from ludwig.api import LudwigModel
  from ludwig.constants import COLUMN, INPUT_FEATURES, PREPROCESSING, SPLIT
  from ludwig.data.split import get_splitter
  from ludwig.features.base_feature import BaseFeature
label_desc: Use image node_id 3 for calling a global function with example usage: get_pred_col(preds, target) and returns: preds, preds