Commit 8707c174 authored by lucas_miranda's avatar lucas_miranda
Browse files

Implemented weight saving callback in model_training.py

parent 672a2925
......@@ -302,7 +302,7 @@ for checkpoint in tqdm(checkpoints):
reconstructions.append(ae.predict(pttest))
print("Done!")
print(predictions)
print([i.shape for i in predictions])
print("Reducing latent space to 2 dimensions for dataviz...")
reducer = LinearDiscriminantAnalysis(n_components=2)
encs = []
......
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