Merge branch 'master' of https://github.com/nylvam/deeplearning-az

joanby · May 17, 2020 · d3f3018 · d3f3018
2 parents ebb6bc1 + f4952ee
commit d3f3018
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/datasets/Part 6 - AutoEncoders (AE)/ae.py b/datasets/Part 6 - AutoEncoders (AE)/ae.py
@@ -106,6 +106,6 @@ def forward(self, x):
         mean_corrector = nb_movies/float(torch.sum(target.data > 0)+1e-10) 
         test_loss += np.sqrt(loss.data*mean_corrector) ## sum(errors) / n_pelis_valoradas
         s += 1.
-print("Test Loss: "+str(train_loss/s))
+print("Test Loss: "+str(test_loss/s))