code: def get_important_features(importance, k):
      return set(np.argsort(importance)[0:k])
  
file_path: imodels/imodels/util/tree_interaction_utils.py
imports: import itertools
  from typing import Set, Tuple
  import numpy
  import pandas
label_desc: Use image node_id 4 for calling a global function with example usage: get_important_features(importance, k) and returns: set