Skip to content

Instantly share code, notes, and snippets.

@e-lin

e-lin/trainMulti2.py

Last active July 9, 2019 12:17
Show Gist options
  • Save e-lin/225cd24d9f1c308517ee3262ec8f55ce to your computer and use it in GitHub Desktop.
Save e-lin/225cd24d9f1c308517ee3262ec8f55ce to your computer and use it in GitHub Desktop.
Download ZIP
AutoEncoder: Training one Autoencoder at a time in multiple graphs
Raw
trainMulti2.py
# first phase of AE, trained on the training data
hidden_output, W1, b1, W4, b4 = train_autoencoder(mnist.train.images, n_neurons=300, n_epochs=4, batch_size=150,
# second phase of AE, trained on the previous Autoencoder's hidden layer output
_, W2, b2, W3, b3 = train_autoencoder(hidden_output, n_neurons=150, n_epochs=4, batch_size=150)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment