Lowercase ROC_AUC

neptune-ai · Jan 15, 2024 · b1b029e · b1b029e
1 parent ea442c8
commit b1b029e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/neptune_sklearn/impl/__init__.py b/src/neptune_sklearn/impl/__init__.py
@@ -231,16 +231,16 @@ 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)
 
         if classification_report:
             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: