tiny update

lazyprogrammer · lazyprogrammer · commit fc06dbce9ea1 · 2017-06-18T17:15:15.000-04:00
diff --git a/unsupervised_class2/rbm_tf.py b/unsupervised_class2/rbm_tf.py
@@ -32,7 +32,6 @@ def build(self, D, M):
 
         # data
         self.X_in = tf.placeholder(tf.float32, shape=(None, D))
-        # self.X_sample_pl = tf.placeholder(tf.float32, shape=(None, D))
 
         # conditional probabilities
         # NOTE: tf.contrib.distributions.Bernoulli API has changed in Tensorflow v1.2
@@ -71,7 +70,7 @@ def build(self, D, M):
             )
         )
 
-    def fit(self, X, learning_rate=0.1, epochs=1, batch_sz=100, show_fig=False):
+    def fit(self, X, epochs=1, batch_sz=100, show_fig=False):
         N, D = X.shape
         n_batches = N // batch_sz
 

Original file line number	Diff line number	Diff line change
`@@ -32,7 +32,6 @@ def build(self, D, M):`
`32`	`32`
`33`	`33`	`# data`
`34`	`34`	`self.X_in = tf.placeholder(tf.float32, shape=(None, D))`
`35`		`- # self.X_sample_pl = tf.placeholder(tf.float32, shape=(None, D))`
`36`	`35`
`37`	`36`	`# conditional probabilities`
`38`	`37`	`# NOTE: tf.contrib.distributions.Bernoulli API has changed in Tensorflow v1.2`
`@@ -71,7 +70,7 @@ def build(self, D, M):`
`71`	`70`	`)`
`72`	`71`	`)`
`73`	`72`
`74`		`- def fit(self, X, learning_rate=0.1, epochs=1, batch_sz=100, show_fig=False):`
	`73`	`+ def fit(self, X, epochs=1, batch_sz=100, show_fig=False):`
`75`	`74`	`N, D = X.shape`
`76`	`75`	`n_batches = N // batch_sz`
`77`	`76`