Skip to content

Commit a472ac9

Browse files
committed
merged changes from tensorflow#25
2 parents 55a34ae + 2ac5f73 commit a472ac9

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

autoencoder/VariationalAutoencoderRunner.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,8 +9,9 @@
99
mnist = input_data.read_data_sets('MNIST_data', one_hot = True)
1010

1111

12-
def minmax_scale(X_train, X_test):
13-
preprocessor = prep.MinMaxScaler(feature_range=(0, 1)).fit(X_train)
12+
13+
def min_max_scale(X_train, X_test):
14+
preprocessor = prep.MinMaxScaler().fit(X_train)
1415
X_train = preprocessor.transform(X_train)
1516
X_test = preprocessor.transform(X_test)
1617
return X_train, X_test
@@ -21,7 +22,7 @@ def get_random_block_from_data(data, batch_size):
2122
return data[start_index:(start_index + batch_size)]
2223

2324

24-
X_train, X_test = minmax_scale(mnist.train.images, mnist.test.images)
25+
X_train, X_test = min_max_scale(mnist.train.images, mnist.test.images)
2526

2627
n_samples = int(mnist.train.num_examples)
2728
training_epochs = 20

0 commit comments

Comments
 (0)