854 B
854 B
Python AutoEncoder from scratch using Numpy
Usage
- Install requirements :
$ pip install -r requirements.txt
- Optionally run mnist_test.py.
$ py mnist_test.py
Training
Instatiate an Autoencoder object :
from autoencoder import Autoencoder
from activations import LeakyReLU
autoencoder = Autoencoder(
[768, 64, 16],
[16, 64, 768],
0.01,
LeakyReLU()
)
And then via the train_dataset method to train over a dataset :
autoencoder.train_dataset(data)
Or via the train to input each data points iteratively :
autoencoder.train(v)
Inference
Use your Autoencoder object with the encode, decode, forward methods like so :
example = ...
code = autoencoder.encode(example)
output = autoencoder.decode(code)
output, code = autoencoder.forward(example)