Kernel Autoencoder

Requirements

• python 3.8.10
• numpy
• scipy
• torch
• torchvision
• tqdm
• ipykernel

pip install -e .

Preparing the MNIST and Fashion MNIST dataset

python src/data_factory/mnist.py --dataset mnist

or

python src/data_factory/mnist.py --dataset fashion_mnist

Train a model

python src/autoencoder/main.py --dataset mnist

or

python src/autoencoder/main.py --dataset fashion_mnist

License

Apache License 2.0