code: def get_class_labels(ids):
      subjs = ["CU", "DA", "DR", "NI", "GU", "IA"]
      class_map = {}
      for i, code in enumerate(subjs):
          with open(
              "/dfs/scratch0/scisurv/clean/{}.tsv".format(code)
          ) as fp:
              fp.readline()
              for line in fp:
                  class_map[int(line.split()[0])] = i
      classes = [class_map[i] for i in ids]
      return classes
  
imports: from __future__ import print_function
  import json
  import numpy
  from networkx.readwrite import json_graph
  from argparse import ArgumentParser
label_desc: Use image node_id 1 for calling a global function with example usage: get_class_labels(ids) and returns: classes