Fixed names

model_selection/test_models.py

@@ -239,7 +239,7 @@ if __name__ == "__main__":
                 roc_display.plot(ax=axes[model_idx][0], lw=2)
                 # Plot the diagonal line for the ROC curve
                 axes[model_idx][0].plot([0, 1], [0, 1], 'k--', lw=2, label='Random Classifier')
-                axes[model_idx][0].set_title(f'ROC Curve for {group}-{method}-{model_name}')
+                axes[model_idx][0].set_title(f'ROC Curve for {group}-{method_names[j]}-{model_name}')
                 axes[model_idx][0].set_xlabel('False Positive Rate')
                 axes[model_idx][0].set_ylabel('True Positive Rate')
                 axes[model_idx][0].legend(loc='lower right')
@@ -251,7 +251,7 @@ if __name__ == "__main__":
                 # Plot the baseline for the PR curve
                 no_skill = len(y_test[y_test == 1]) / len(y_test)
                 axes[model_idx][1].plot([0, 1], [no_skill, no_skill], 'k--', lw=2, label='No Skill')
-                axes[model_idx][1].set_title(f'PR Curve for {group}-{method}-{model_name}')
+                axes[model_idx][1].set_title(f'PR Curve for {group}-{method_names[j]}-{model_name}')
                 axes[model_idx][1].set_xlabel('Recall')
                 axes[model_idx][1].set_ylabel('Precision')
                 axes[model_idx][1].legend(loc='lower left')
@@ -269,7 +269,7 @@ if __name__ == "__main__":
                 cax = divider.append_axes("right", size="5%", pad=0.05)
                 plt.colorbar(cmp.im_, cax=cax)
 
-                axes[model_idx][2].set_title(f'CM for {group}-{method}-{model_name}')
+                axes[model_idx][2].set_title(f'CM for {group}-{method_names[j]}-{model_name}')
                 axes[model_idx][2].set_xlabel('Predicted label')
                 axes[model_idx][2].set_ylabel('True label')
                 # ----------------------------------------------------------