basketballcantho/CSIROBoeingPhase5-Vietnam
1
0
Fork 0
mirror of https://git.victorphan.net/basketballcantho/CSIROBoeingPhase5-Vietnam.git synced 2026-08-05 05:43:10 +07:00
Code Issues Packages Projects Releases Wiki Activity

update 01

Browse Source
This commit is contained in:
Victor Phan
2025-11-13 19:49:16 +07:00
parent 92acf224ff
commit 71daf2a704
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
01.train_ODC_XGBoost.ipynb
+4 -1
View File
@@ -3000,10 +3000,13 @@
"from sklearn.metrics import ConfusionMatrixDisplay\n", "from sklearn.metrics import ConfusionMatrixDisplay\n",
"import matplotlib.pyplot as plt\n", "import matplotlib.pyplot as plt\n",
"\n", "\n",
"# Create figure first\n",
"fig, ax = plt.subplots(figsize=(10, 8))\n",
"\n",
"class_names = list(label_mapping.keys())\n", "class_names = list(label_mapping.keys())\n",
"cm = confusion_matrix(y_test_np, y_pred_test)\n", "cm = confusion_matrix(y_test_np, y_pred_test)\n",
"disp = ConfusionMatrixDisplay(confusion_matrix=cm, display_labels=class_names)\n", "disp = ConfusionMatrixDisplay(confusion_matrix=cm, display_labels=class_names)\n",
"disp.plot(figsize=(10, 8), cmap='Blues')\n", "disp.plot(cmap='Blues', ax=ax)\n",
"plt.xticks(rotation=45, ha='right')\n", "plt.xticks(rotation=45, ha='right')\n",
"plt.title('XGBoost Confusion Matrix')\n", "plt.title('XGBoost Confusion Matrix')\n",
"plt.tight_layout()\n", "plt.tight_layout()\n",