ajoubrel-ensae/BDC-team-1
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

adjust font size

Browse Source
This commit is contained in:
Alexis REVELLE 2024-04-04 11:46:15 +00:00
parent 1ebb83e3c4
commit f5b6075431
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
utils_ml.py
View File

@ -161,7 +161,7 @@ def preprocess(type_of_model, type_of_activity):
def draw_confusion_matrix(y_test, y_pred, model): def draw_confusion_matrix(y_test, y_pred, model):
conf_matrix = confusion_matrix(y_test, y_pred) conf_matrix = confusion_matrix(y_test, y_pred)
sns.heatmap(conf_matrix, annot=True, fmt='d', cmap='Blues', xticklabels=['Class 0', 'Class 1'], yticklabels=['Class 0', 'Class 1']) sns.heatmap(conf_matrix, annot=True, fmt='d', cmap='Blues', xticklabels=['Class 0', 'Class 1'], yticklabels=['Class 0', 'Class 1'], annot_kws={"size": 14})
plt.xlabel('Predicted') plt.xlabel('Predicted')
plt.ylabel('Actual') plt.ylabel('Actual')
plt.title('Confusion Matrix') plt.title('Confusion Matrix')
@ -180,10 +180,10 @@ def draw_roc_curve(X_test, y_pred_prob, model):
plt.plot(fpr, tpr, label="ROC curve(area = %0.3f)" % roc_auc) plt.plot(fpr, tpr, label="ROC curve(area = %0.3f)" % roc_auc)
plt.plot([0, 1], [0, 1], color="red",label="Random Baseline", linestyle="--") plt.plot([0, 1], [0, 1], color="red",label="Random Baseline", linestyle="--")
plt.grid(color='gray', linestyle='--', linewidth=0.5) plt.grid(color='gray', linestyle='--', linewidth=0.5)
plt.xlabel("False Positive Rate") plt.xlabel("False Positive Rate", fontsize=14)
plt.ylabel("True Positive Rate") plt.ylabel("True Positive Rate", fontsize=14)
plt.title("ROC Curve", size=18) plt.title("ROC Curve", size=18)
plt.legend(loc="lower right") plt.legend(loc="lower right", fontsize=14)
plt.show() plt.show()
save_file_s3("Roc_curve_", type_of_activity, type_of_model, model) save_file_s3("Roc_curve_", type_of_activity, type_of_model, model)