Fixed wrong score function name in deep autoencoders.

gabrieleangeletti · Apr 14, 2017 · 6078fb6 · 6078fb6
1 parent b254e9e
commit 6078fb6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/cmd_line/autoencoders/run_stacked_autoencoder_unsupervised.py b/cmd_line/autoencoders/run_stacked_autoencoder_unsupervised.py
@@ -176,7 +176,7 @@ def load_params_npz(npzfilepath):
     sdae.fit(trX, trRef, vlX, vlRef)
 
     # Compute the reconstruction loss of the model
-    print('Test set reconstruction loss: {}'.format(sdae.compute_reconstruction_loss(teX, teRef)))
+    print('Test set reconstruction loss: {}'.format(sdae.score(teX, teRef)))
 
     # Save the predictions of the model
     if FLAGS.save_reconstructions:

diff --git a/cmd_line/boltzmann/run_deep_autoencoder.py b/cmd_line/boltzmann/run_deep_autoencoder.py
@@ -152,7 +152,7 @@ def load_params_npz(npzfilepath):
     srbm.fit(trX, trRef, vlX, vlRef)
 
     # Compute the reconstruction loss of the model
-    print('Test set reconstruction loss: {}'.format(srbm.compute_reconstruction_loss(teX, teRef)))
+    print('Test set reconstruction loss: {}'.format(srbm.score(teX, teRef)))
 
     # Save the predictions of the model
     if FLAGS.save_reconstructions: