From cd0cc05f8e46299372aaa0978637d0c18a5c3424 Mon Sep 17 00:00:00 2001 From: Anirudh Dagar Date: Tue, 2 Apr 2024 18:13:08 +0200 Subject: [PATCH] DOC: Mention quadratic_kappa metric in eval_metric docs (#3996) --- core/src/autogluon/core/models/abstract/abstract_model.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/core/src/autogluon/core/models/abstract/abstract_model.py b/core/src/autogluon/core/models/abstract/abstract_model.py index 755f9e58378..9628f4f03b6 100644 --- a/core/src/autogluon/core/models/abstract/abstract_model.py +++ b/core/src/autogluon/core/models/abstract/abstract_model.py @@ -85,9 +85,9 @@ class AbstractModel: If `eval_metric = None`, it is automatically chosen based on `problem_type`. Defaults to 'accuracy' for binary and multiclass classification and 'root_mean_squared_error' for regression. Otherwise, options for classification: - ['accuracy', 'balanced_accuracy', 'f1', 'f1_macro', 'f1_micro', 'f1_weighted', - 'roc_auc', 'roc_auc_ovo_macro', 'average_precision', 'precision', 'precision_macro', 'precision_micro', - 'precision_weighted', 'recall', 'recall_macro', 'recall_micro', 'recall_weighted', 'log_loss', 'pac_score'] + ['accuracy', 'balanced_accuracy', 'f1', 'f1_macro', 'f1_micro', 'f1_weighted', 'roc_auc', 'roc_auc_ovo_macro', + 'average_precision', 'precision', 'precision_macro', 'precision_micro', 'precision_weighted', 'recall', + 'recall_macro', 'recall_micro', 'recall_weighted', 'log_loss', 'pac_score', 'quadratic_kappa'] Options for regression: ['root_mean_squared_error', 'mean_squared_error', 'mean_absolute_error', 'median_absolute_error', 'r2'] Options for quantile regression: