All models running.

2019-08-24 19:05:46 +02:00
parent 7b0b96eaa3
commit 18305a9e7e
6 changed files with 45 additions and 45 deletions
--- a/viz/viz_latent.py
+++ b/viz/viz_latent.py
@ -62,7 +62,7 @@ def load_and_predict(path_like_element):
        plot.savefig(os.path.join(os.path.dirname(path_like_element), f'latent_space_{idx}.png'))


-def viz_latent(prediction, title=f'Latent Space '):
+def viz_latent(prediction):
    if prediction.shape[-1] <= 1:
        raise ValueError('How did this happen?')
    elif prediction.shape[-1] == 2: