HomeDocsAPI Reference
Kumo.ai
Docs

tune_metric

tune_metric: (auroc | acc | mae | f1 | loss | ap_macro) (Optional)

Description

The tune metric specifies the key metric that AutoML uses to optimize training experiments. This metric plays a critical role in:

  1. Early Stopping: During training, Kumo monitors the tune metric to decide when to stop training models that are unlikely to improve further. Learn more about this feature in the early_stopping documentation.
  2. Model Selection: Among all models trained during AutoML, Kumo selects the best-performing model based on the tune metric. This selected model is then evaluated on the test set for final validation.

The tune metric needs to be in the list of metrics in the model plan. For available metrics, please refer to metrics and Kumo Evaluation Metrics.

Supported Task Types

  • All

Default Values

Task TypeDefault Value
Binary Classificationauroc
Multiclass Classificationacc
Multilabel Classificationap_macro
Regressionmae
Forecastingmae
Temporal Link Predictionmap@10
Static Link Predictionmap@10
Multilabel Rankingmap@10