Merge branch 'main' of github.com:UBC-MDS/DSCI-522-2425-team35-Heart_…

…disease_diagnostic_machine into quarto_report
UBC-MDS · Dec 8, 2024 · 048f2fe · 048f2fe
2 parents 632bab5 + 7e759f6
commit 048f2fe
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/results/figures/confusion_matrix.png b/results/figures/confusion_matrix.png
diff --git a/scripts/5_evaluate.py b/scripts/5_evaluate.py
@@ -73,7 +73,13 @@ def main(train, test, write_to):
 
     # Save confusion matrix
     confmat = ConfusionMatrixDisplay.from_estimator(best_model, X_test, y_test, values_format="d")
-    confmat.figure_.savefig(os.path.join(write_to, "figures", "confusion_matrix.png"))
+    confmat.figure_.set_size_inches(10, 7)  # Set custom figure size
+    confmat.figure_.tight_layout()
+    confmat.figure_.savefig(
+        os.path.join(write_to, "figures", "confusion_matrix.png"),
+        bbox_inches='tight'
+    )
+
 
 if __name__ == '__main__':
     main()