Skip to content

GitLab

Commit dec7c89f authored by Joshua Howard's avatar Joshua Howard Committed by Neal Wu
Browse files

Variational Autoencoder generate() function fixed (z fed in rather than z_mean)

parent ae50fa99
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 3 deletions
autoencoder/autoencoder_models/VariationalAutoencoder.py
View file @ dec7c89f
import tensorflow as tf import tensorflow as tf
import numpy as np
class VariationalAutoencoder(object): class VariationalAutoencoder(object):
...@@ -57,8 +56,8 @@ class VariationalAutoencoder(object): ...@@ -57,8 +56,8 @@ class VariationalAutoencoder(object):
def generate(self, hidden = None): def generate(self, hidden = None):
if hidden is None: if hidden is None:
hidden = np.random.normal(size=self.weights["b1"]) hidden = self.sess.run(tf.random_normal([1, self.n_hidden]))
return self.sess.run(self.reconstruction, feed_dict={self.z_mean: hidden}) return self.sess.run(self.reconstruction, feed_dict={self.z: hidden})
def reconstruct(self, X): def reconstruct(self, X):
return self.sess.run(self.reconstruction, feed_dict={self.x: X}) return self.sess.run(self.reconstruction, feed_dict={self.x: X})
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment