From 5d4eaf6e28d065c9b73a6a4dd58e7e8768efe7f8 Mon Sep 17 00:00:00 2001 From: Stefan Date: Fri, 22 Apr 2022 10:22:48 +0200 Subject: [PATCH] improved froc visualisation --- scripts/5.Visualize_frocs.py | 26 ++++++++++++++++---------- 1 file changed, 16 insertions(+), 10 deletions(-) diff --git a/scripts/5.Visualize_frocs.py b/scripts/5.Visualize_frocs.py index 7bb6da6..5f4a96f 100755 --- a/scripts/5.Visualize_frocs.py +++ b/scripts/5.Visualize_frocs.py @@ -28,21 +28,27 @@ print(experiments) experiment_path = [] experiment_metrics = {} auroc = [] + +fig = plt.figure(1) +ax = fig.add_subplot(111) +for idx in range(len(args.experiment)): + experiment_path = f'./../train_output/{experiments[idx]}/froc_metrics_focal_10.yml' + experiment_metrics = read_yaml_to_dict(experiment_path) + + plt.plot(experiment_metrics["FP_per_case"], experiment_metrics["sensitivity"],color=colors[idx],linestyle=plot_type[idx]) + ax.set(xscale="log") + ax.axes.xaxis.set_minor_locator(tkr.LogLocator(base=10, subs='all')) + ax.axes.xaxis.set_minor_formatter(tkr.NullFormatter()) + ax.axes.xaxis.set_major_formatter(tkr.ScalarFormatter()) + ax.axes.grid(True, which="both", ls="--", c='#d3d3d3') + ax.axes.set_xlim(left=0, right=3) + ax.axes.xaxis.set_major_locator(tkr.FixedLocator([0,0.1,1,3])) + for idx in range(len(args.experiment)): experiment_path = f'./../train_output/{experiments[idx]}/froc_metrics_focal_10.yml' experiment_metrics = read_yaml_to_dict(experiment_path) auroc.append(round(experiment_metrics['auroc'],3)) - fig1, ax1 = plt.subplots(1,1) - ax1.plot(experiment_metrics["FP_per_case"], experiment_metrics["sensitivity"],color=colors[idx],linestyle=plot_type[idx]) - ax1.set(xscale="log") - ax1.xaxis.set_minor_locator(tkr.LogLocator(base=10, subs='all')) - ax1.xaxis.set_minor_formatter(tkr.NullFormatter()) - ax1.xaxis.set_major_formatter(tkr.ScalarFormatter()) - ax1.grid(True, which="both", ls="--", c='#d3d3d3') - ax1.set_xlim(left=0, right=3) - ax1.xaxis.set_major_locator(tkr.FixedLocator([0,0.1,1,3])) - plt.figure(2) plt.plot(experiment_metrics["fpr"], experiment_metrics["tpr"],color=colors[idx],linestyle=plot_type[idx])