From b1b029ec0023ad89bdc2321d0e509f8fa3eff949 Mon Sep 17 00:00:00 2001 From: Siddhant Sadangi Date: Mon, 15 Jan 2024 10:33:50 +0100 Subject: [PATCH] Lowercase ROC_AUC --- src/neptune_sklearn/impl/__init__.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/neptune_sklearn/impl/__init__.py b/src/neptune_sklearn/impl/__init__.py index 85d2c7e..e881975 100644 --- a/src/neptune_sklearn/impl/__init__.py +++ b/src/neptune_sklearn/impl/__init__.py @@ -231,7 +231,7 @@ def create_classifier_summary(classifier, X_train, X_test, y_train, y_test, nrow if log_charts: classification_report = create_classification_report_chart(classifier, X_train, X_test, y_train, y_test) confusion_matrix = create_confusion_matrix_chart(classifier, X_train, X_test, y_train, y_test) - ROC_AUC = create_roc_auc_chart(classifier, X_train, X_test, y_train, y_test) + roc_auc = create_roc_auc_chart(classifier, X_train, X_test, y_train, y_test) precision_recall = create_precision_recall_chart(classifier, X_test, y_test) class_prediction_error = create_class_prediction_error_chart(classifier, X_train, X_test, y_train, y_test) @@ -239,8 +239,8 @@ def create_classifier_summary(classifier, X_train, X_test, y_train, y_test, nrow cls_summary["diagnostics_charts/classification_report"] = classification_report if confusion_matrix: cls_summary["diagnostics_charts/confusion_matrix"] = confusion_matrix - if ROC_AUC: - cls_summary["diagnostics_charts/ROC_AUC"] = ROC_AUC + if roc_auc: + cls_summary["diagnostics_charts/ROC_AUC"] = roc_auc if precision_recall: cls_summary["diagnostics_charts/precision_recall"] = precision_recall if class_prediction_error: