cosmetic adjustments

joshuawe · Nov 10, 2023 · fe31ef7 · fe31ef7
1 parent 9d7d888
commit fe31ef7
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/plotsandgraphs/binary_classifier.py b/plotsandgraphs/binary_classifier.py
@@ -66,7 +66,7 @@ def plot_confusion_matrix(
         values_format="",
         colorbar=False,
         ax=ax,
-        text_kw={"visible": False},
+        # text_kw={"visible": False},
     )
     cmd.texts_ = []
     cmd.text_ = []
@@ -115,7 +115,7 @@ def plot_confusion_matrix(
 
 
 def plot_classification_report(
-    y_test: np.ndarray,
+    y_true: np.ndarray,
     y_pred: np.ndarray,
     title="Classification Report",
     figsize=(8, 4),
@@ -158,7 +158,7 @@ def plot_classification_report(
 
     cmap = "YlOrRd"
 
-    clf_report = classification_report(y_test, y_pred, output_dict=True, **kwargs)
+    clf_report = classification_report(y_true, y_pred, output_dict=True, **kwargs)
     keys_to_plot = [
         key
         for key in clf_report.keys()