code: def _get_pred_int(self, pred_score):
      if len(pred_score.shape) == 1 or pred_score.shape[1] == 1:
          return (pred_score > cfg.model.thresh).long()
      else:
          return pred_score.max(dim=1)[1]
  
imports: import logging
  import math
  import sys
  import torch
  import graphgym.register
  from graphgym.config import cfg
  from graphgym.utils.device import get_current_gpu_usage
  from graphgym.utils.io import dict_to_json, dict_to_tb, makedirs
label_desc: Use image node_id 7 for calling the Logger obj's underlying member method code with example usage: obj._get_pred_int(pred_score) and returns: unknown, pred_score