Skip to content
Snippets Groups Projects
Select Git revision
  • 28a906cef3a8dbf0454166226938e85db0b6b8be
  • master default protected
2 results

visualize.py

Blame
    • visualize.py 1020 B
    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 
    

    

    

    import numpy as np
import matplotlib.pyplot as plt
import train_URWF
import train_UIRWF
import params

def plot():
    plt.rcParams['text.usetex'] = True
    plt.title("err vs epoch index")
    plt.xlabel("$epochs$")
    plt.ylabel(r'$\min \| z_{i}e^{-i\phi} - x_i \|_F$', fontsize=14, color='k')
    plt.figure(1)
    plt.semilogy(np.arange(0, params.EPOCHS), train_URWF.err_URWF_train, color ="red",label="URWF on train")
    plt.semilogy(np.arange(0, params.EPOCHS), train_URWF.err_URWF_val, color ="blue",label="URWF on test")
    plt.semilogy(np.arange(0, params.EPOCHS), train_UIRWF.err_UIRWF_train, color ="orange",label="UIRWF on train")
    plt.semilogy(np.arange(0, params.EPOCHS), train_UIRWF.err_UIRWF_val, color ="green",label="UIRWF on test")
    plt.title(r'URWF vs UIRWF on train/test data',fontsize=10)
    plt.legend()
    plt.savefig('Scenario = '+str(params.scenario)+', Layers = '+str(params.T)+', Epochs = '+str(params.EPOCHS)+'_ LR = '+str(params.LR)+'_Learning_Process.pdf')
    # plt.show()