Deep Clustering with Convolutional Autoencoders (DCEC)
Keras implementation for ICONIP-2017 paper:
- Xifeng Guo, Xinwang Liu, En Zhu, Jianping Yin. Deep Clustering with Convolutional Autoencoders. ICONIP 2017.
Usage
-
Install Keras >=v2.0, scikit-learn and git
sudo pip install keras scikit-learn
sudo apt-get install git
-
Clone the code to local.
git clone https://github.com/XifengGuo/DCEC.git DCEC
-
Prepare datasets.
cd DCEC/data/usps bash ./download_usps.sh cd ../..
-
Run experiment on MNIST.
python DCEC.py mnist
The DCEC model will be saved to "results/temp/dcec_model_final.h5". -
Run experiment on USPS.
python DCEC.py usps
-
Run experiment on MNIST-TEST.
python DCEC.py mnist-test