diff --git a/cmd_line/autoencoders/run_stacked_autoencoder_unsupervised.py b/cmd_line/autoencoders/run_stacked_autoencoder_unsupervised.py index e33b844..73cdf36 100644 --- 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 index e5de2d3..c9e6321 100644 --- 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: