feat: loss method + mv data reshaping out of Autoencoder class

This commit is contained in:
Lenoctambule
2026-03-28 02:40:10 +01:00
parent e5520bf050
commit 6155649655
3 changed files with 25 additions and 11 deletions

View File

@@ -25,10 +25,10 @@ def regularize(v: np.ndarray) -> np.ndarray:
return (v - v_min) / (v_max - v_min)
def dynamic_loss_plot_init():
def dynamic_loss_plot_init(losses: list):
plt.ion()
fig, ax = plt.subplots()
line, = ax.plot([], [], label="Loss")
line, = ax.plot([0], losses, label="Loss")
ax.set_xlabel("Epoch")
ax.set_ylabel("Loss")
ax.set_title("Training Loss")