feat: custom load_mnist function and rm useless dependency

This commit is contained in:
Lenoctambule
2026-03-27 21:54:17 +01:00
parent fa2bfe4ef5
commit 32e27b4b62
3 changed files with 16 additions and 6 deletions

View File

@@ -91,7 +91,6 @@ class Autoencoder:
while True:
print(
f"{LOADER[epoch % len(LOADER)]} Training \t({epoch=} error={prev_error:.2f})", # noqa
end="\r"
)
error = 0
for x in data_set:
@@ -113,7 +112,7 @@ class Autoencoder:
epoch += 1
if display_loss is True:
dynamic_loss_plot_finish(ax, line)
print("\r#Training complete !")
print("#Training complete !")
return losses
def encode(self, v: np.ndarray) -> np.ndarray: