code: def profile_NK_trade_off(dataset):
      """
      This is get from the profling result.
      We try various N/K combinations, and find this is better.
      """
      if dataset == Config.c10:
          return 85
      elif dataset == Config.c100:
          return 85
      elif dataset == Config.imgNet:
          return 130
      else:
          return 30
  
file_path: singa/examples/model_selection/TRAILS-Database-Native-Model-Selection/internal/ml/model_selection/src/query_api/interface.py
imports: from src.common.constant import Config, CommonVars
  from src.query_api.query_api_img import Gt201, Gt101
  from src.query_api.query_api_mlp import GTMLP
  from src.query_api.query_api_img import ImgScoreQueryApi
  from typing import *
label_desc: Use image node_id 1 for calling a global function with example usage: profile_NK_trade_off(dataset) and returns: int, int, int, int