diff --git a/docs/assets/roc_auc_curve.png b/docs/assets/roc_auc_curve.png index bc2f71f..81cc2f6 100644 Binary files a/docs/assets/roc_auc_curve.png and b/docs/assets/roc_auc_curve.png differ diff --git a/src/processing.py b/src/processing.py index 91a5b0d..f648096 100644 --- a/src/processing.py +++ b/src/processing.py @@ -98,15 +98,15 @@ def plot_all_figures(results): plot_individual_figure( results, type="roc", - x_axis="fpr", - y_axis="tpr", + x_axis="False positive rate", + y_axis="True positive rate", fig_title="ROC AUC curve", ) plot_individual_figure( results, type="confusion_matrix", - x_axis="fpr", - y_axis="tpr", + x_axis="Predicted values", + y_axis="Real values", fig_title="Confusion Matrix", )