Refactored a bit the old visualization pipeline - added umap to requirements

079198bb · lucas_miranda · 6633251a · 079198bb
Commit 079198bb authored Nov 20, 2020 by lucas_miranda
--- a/examples/visualizations/train_viz_data_generator.py
+++ b/examples/visualizations/train_viz_data_generator.py
@@ -377,7 +377,7 @@ reconstructions = []
    predictor=predictor,
    kl_warmup_epochs=10,
    mmd_warmup_epochs=10,
-    **hparams
+    architecture_hparams=hparams
 ).build(pttest.shape)
 gmvaep.build(pttest.shape)