konlyi / Dual-aae

Deep Clustering Learning

This repository provides a PyTorch implementation of the paper Dual Adversarial Autoencoders for Clustering.

Dependencies

Python 3.5

Pytorch 0.2.0

Usage

1.Training

To train Dual-AAE on MNIST, run the training script below.

$ python main.py

2.Testing

To test Dual-AAE on MNIST.

$ python mnist_test.py