Skip to content

GitLab

"docs/source/api/vscode:/vscode.git/clone" did not exist on "1e84168edd695992d983992997ad02008181c840"
Commit 4874c96a authored by Jiří Vahala's avatar Jiří Vahala
Browse files

VAE runner fixed, data needs different scaler

parent 70702f79
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
autoencoder/VariationalAutoencoderRunner.py
View file @ 4874c96a
...@@ -9,8 +9,8 @@ from autoencoder.autoencoder_models.VariationalAutoencoder import VariationalAut ...@@ -9,8 +9,8 @@ from autoencoder.autoencoder_models.VariationalAutoencoder import VariationalAut
mnist = input_data.read_data_sets('MNIST_data', one_hot = True) mnist = input_data.read_data_sets('MNIST_data', one_hot = True)
def standard_scale(X_train, X_test): def min_max_scale(X_train, X_test):
preprocessor = prep.StandardScaler().fit(X_train) preprocessor = prep.MinMaxScaler().fit(X_train)
X_train = preprocessor.transform(X_train) X_train = preprocessor.transform(X_train)
X_test = preprocessor.transform(X_test) X_test = preprocessor.transform(X_test)
return X_train, X_test return X_train, X_test
...@@ -21,7 +21,7 @@ def get_random_block_from_data(data, batch_size): ...@@ -21,7 +21,7 @@ def get_random_block_from_data(data, batch_size):
return data[start_index:(start_index + batch_size)] return data[start_index:(start_index + batch_size)]
X_train, X_test = standard_scale(mnist.train.images, mnist.test.images) X_train, X_test = min_max_scale(mnist.train.images, mnist.test.images)
n_samples = int(mnist.train.num_examples) n_samples = int(mnist.train.num_examples)
training_epochs = 20 training_epochs = 20
......
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