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()