plot.py 333 Bytes
Newer Older
Eva Lina Fesefeldt's avatar
Eva Lina Fesefeldt committed
1 2 3 4 5 6 7 8 9 10 11 12
import numpy as np
import matplotlib.pyplot as plt

loss_vs_hidden_layer = np.load("results/mini_loss_vs_size_hidden_layer.npy")

for size_hidden_layer in range(0,51, 5):
    plt.semilogy(loss_vs_hidden_layer[size_hidden_layer,:], label="n="+str(size_hidden_layer+1))

plt.xlabel("Epoche")
plt.ylabel("Loss")
plt.legend()
plt.show()