plot_utils.py

import matplotlib.pyplot as plt
import numpy as np

plt.rcParams.update({'font.size': 18})
plt.rc('font', family='serif')
plt.style.use('seaborn-muted')

def plot_results(list_results):

    Name_tilte = list_results[0]['dataset'] + " Autoencoder"

    ColorsList = ['b', 'r', 'c', 'g', 'y', 'k', 'm', 'brown']

    for i in range(len(list_results)):
        list_results[i]['Color'] = ColorsList[i]

    fig, ax = plt.subplots(figsize = (10,10))
    fig.patch.set_facecolor('white')

    for dict_save in list_results:
        ax.plot(dict_save['timesCPU'], dict_save['train_losses'], label= dict_save['algorithm'], color = dict_save['Color'], linewidth=3)

    # ax.grid(True)
    lgd = plt.legend(frameon=True, loc = 'upper right', framealpha = 1, edgecolor = 'black', fancybox = False)
    lgd.get_frame().set_linewidth(1.0)
    for line in lgd.get_lines():
        line.set_linewidth(3.0)

    handles, labels = ax.get_legend_handles_labels()
    ax.legend(reversed(handles), reversed(labels))

    ax.set_ylabel('Training loss')
    ax.set_xlabel('CPU time')

    ax.spines['top'].set_visible(True)
    ax.spines['right'].set_visible(True)

    ax.xaxis.set_tick_params(top='on', direction='in', width=2)
    ax.yaxis.set_tick_params(right='on', direction='in', width=2)

    plt.title(Name_tilte, fontsize = 25, fontweight='normal')
    plt.yscale("log")
    plt.show()


    fig, ax = plt.subplots(figsize = (10,10))
    fig.patch.set_facecolor('white')

    for dict_save in list_results:
        ax.plot(dict_save['epochs'], dict_save['train_losses'], label= dict_save['algorithm'], color = dict_save['Color'], linewidth=3)

    # ax.grid(True)
    lgd = plt.legend(frameon=True, loc = 'upper right', framealpha = 1, edgecolor = 'black', fancybox = False)
    lgd.get_frame().set_linewidth(1.0)
    for line in lgd.get_lines():
        line.set_linewidth(3.0)

    handles, labels = ax.get_legend_handles_labels()
    ax.legend(reversed(handles), reversed(labels))

    ax.set_ylabel('Training loss')
    ax.set_xlabel('Epochs')

    ax.spines['top'].set_visible(True)
    ax.spines['right'].set_visible(True)

    ax.xaxis.set_tick_params(top='on', direction='in', width=2)
    ax.yaxis.set_tick_params(right='on', direction='in', width=2)

    plt.title(Name_tilte, fontsize = 25, fontweight='normal')
    plt.yscale("log")
    plt.show()