Plot.py

import matplotlib.pyplot as plt
import numpy as np

n_folds = 5
ds_name = [["AD","NC"],["AD","EMCI"],["AD","LMCI"],["LMCI","NC"],["LMCI","EMCI"],["EMCI","NC"]]
plot_history_path="H:\Google Drive\plot"
plot_data_type= ["ROC_AUC", "evaluation"]

def get_plot_data(history_path, prefix, plot_data_type):

    whole_history_list = list()
    whole_data_list = list()
    history_list = list()
    data_list = list()
    history_txt_prefix="{}\\{}_history_{}_vs_{}".format(str(history_path),str(plot_data_type),str(prefix[0]),str(prefix[1]))
    for j in range(1,n_folds+1):
        history_txt="{}_fold_index_{}.txt".format(str(history_txt_prefix),str(j))
        history_list.append(history_txt)
        #print(history_txt)
        data = np.loadtxt(history_txt, delimiter=',', usecols = 1)
        data_list.append(data)
    whole_history_list.append(history_list)
    whole_data_list.append(data_list)
    return np.asanyarray(whole_data_list)

def plot_ROC_AUC(plot_history_path,prefix,plot_data_type):
    mean = list()
    std = list()
    ROC_AUC = get_plot_data(plot_history_path, prefix, plot_data_type)
    #for i in range(ROC_AUC.shape[-1]):
    for i in range(70):
        mean.append(np.mean(ROC_AUC[:, :, i]))
        std.append(np.std(ROC_AUC[:, :, i]))
    mean = np.asanyarray(mean)
    std = np.asanyarray(std)

    plt.plot(mean)
    plt.fill_between(list(range(len(mean))), mean + std, mean - std, color='gray', alpha=0.2)
    #plt.title('AD vs NC classification ROC AUC')
    plt.title('{} vs {} classification ROC AUC'.format(str(prefix[0]), str(prefix[1])))
    plt.ylabel('ROC AUC')
    plt.xlabel('epoch')
    plt.legend(['Validation ROC AUC mean', 'Validation ROC AUC std'], loc='lower left')
    plt.show()

def calculate_results(results):
    mean = list()
    std = list()
    mean.append(np.mean(results[:, :, -1]))
    std.append(np.std(results[:, :, -1]))
    mean = np.asanyarray(mean)
    std = np.asanyarray(std)
    return mean, std

def print_evaluation_results(plot_data_type):
    mean_list = list()
    std_list = list()
    for i in range(len(ds_name)):
        results = get_plot_data(plot_history_path, ds_name[i], plot_data_type)
        mean, std = calculate_results(results)
        mean_list.append(mean)
        std_list.append(std)
    mean_list = np.asanyarray(mean_list)
    std_list = np.asanyarray(std_list)
    if plot_data_type == "ROC_AUC" :
        title = plot_data_type
    else :
        title = "{} acc".format(str(plot_data_type))
    print("\n    {} : [mean] ± [std]\n".format(str(title)))
    for i in range(len(ds_name)):
        print("{} vs {} : {} ± {}".format(str(ds_name[i][0]),str(ds_name[i][1]),mean_list[i],std_list[i]))

plot_ROC_AUC(plot_history_path, ds_name[0], plot_data_type[0])
print_evaluation_results(plot_data_type[0])
print_evaluation_results(plot_data_type[1])