def extract_model_specs()

in tfx_addons/xgboost_evaluator/xgboost_predict_extractor.py [0:0]

• 8 lines of code
• 4 McCabe index (conditional complexity)

  def extract_model_specs(self):
    label_specs = {}
    for config in self._eval_config.model_specs:
      if config.name:
        label_specs[config.name] = config.label_key
      else:  # if the input name to ModelSpec is None, ModelSpec doesn't save it and config.name resolves to ''.
        label_specs[None] = config.label_key
    return label_specs