Commit 6c1e2b7f authored by lucas_miranda's avatar lucas_miranda
Browse files

Implemented weight saving callback in model_training.py

parent e8576600
......@@ -349,6 +349,10 @@ dfencs["epoch"] = np.array(
)
dfencs.columns = ["x", "y", "cluster", "epoch"]
dfencs["trajectories"] = np.tile(pttest[:, 6, 1], len(checkpoints) + 1)
print(trajectories.shape)
print(np.concatenate(reconstructions).shape)
dfencs["reconstructions"] = np.concatenate(reconstructions)[:, 6, 1]
# Cluster membership animated over epochs
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment