re run with cm

model_selection/test_models.py

@@ -272,7 +272,7 @@ if __name__ == "__main__":
                 # Get confusion matrix plot
                 y_pred = model.predict(X_test)
                 cm = confusion_matrix(y_test, y_pred)
-                ConfusionMatrixDisplay(cm).plot(ax=axes[model_idx[2]])
+                ConfusionMatrixDisplay(cm).plot(ax=axes[model_idx][2])
                 # Give name to plots
                 axes[model_idx][0].set_title(f'ROC Curve for {model_name}')
                 axes[model_idx][1].set_title(f'PR Curve for {model_name}')