StatisticsMaster/box_plot.py at master · oysteiae/StatisticsMaster · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
import matplotlib.pyplot as plt
import numpy as np
import csv

def read_scores(path_to_file, index):
    scores = []

    with open(path_to_file, 'r') as log:
        log = csv.reader(log, delimiter='\t')
        first = True

        for row in log:
            if(first):
                first = False
                continue
            scores.append(float(row[index]))

    return scores

def box_plot(path_to_scores_cnn, path_to_scores_unet, path_to_scores_deepmedic, index):
    bottom = 0.7
    top = 1.0

    cnn_scores = read_scores(path_to_scores_cnn, index)
    unet_scores = read_scores(path_to_scores_unet, index)
    deepmedic_scores = read_scores(path_to_scores_deepmedic, index)

    data_to_plot = [cnn_scores, unet_scores, deepmedic_scores]

    fig = plt.figure(1, figsize=(9, 6))
    # Create an axes instance
    ax = fig.add_subplot(111)
    # Create the boxplot
    bp = ax.boxplot(data_to_plot)
    ax.set_ylabel("Dice score", fontsize=12)
    ax.set_xlabel("Architecture", fontsize=12)

    architectures = ["CNN", "U-Net", "DeepMedic"]
    ax.set_ylim(bottom, top)
    ax.set_xticklabels(architectures,
                        rotation=45, fontsize=15)

    plt.gcf().subplots_adjust(bottom=0.22)
    plt.show()
    fig.savefig("D:\\Master\\Graphs\\BoxplotTrainedOnOne.png")

# box_plot( "D:\\Master\\DefinitiveFinalExperiments\\CNNAllDataNotResampled\\testing_indices_CNNAllDataNotResampled_scores.tsv", "D:\\Master\\DefinitiveFinalExperiments\\UnetFinalDataNotResampled\\testing_indices_UnetFinalDataNotResampled_scores.tsv", "D:\\Master\\deepmedicexperiments\\DeepMedicAllDataNotResampled_scores.tsv", 1)
# box_plot( "D:\\Master\\DefinitiveFinalExperiments\\CNN40FromAll\\testing_indices_CNN40FromAll_scores.tsv", "D:\\Master\\DefinitiveFinalExperiments\\Unet40FromAll\\testing_indices_Unet40FromAll_scores.tsv", "D:\\Master\\deepmedicexperiments\\DeepMedic40FromAll_scores.tsv", 1)
box_plot( "D:\\Master\\deepmedicexperiments\\combined_scores\\CNNScoresOne.txt", "D:\\Master\\deepmedicexperiments\\combined_scores\\UnetScoresOne.txt", "D:\\Master\\deepmedicexperiments\\combined_scores\\DeepMedicAllScoresOne.txt", 1)