/DCEC

Primary LanguagePythonMIT LicenseMIT

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

  1. Install Keras >=v2.0, scikit-learn and git
    sudo pip install keras scikit-learn
    sudo apt-get install git

  2. Clone the code to local.
    git clone https://github.com/XifengGuo/DCEC.git DCEC

  3. Prepare datasets.

     cd DCEC/data/usps   
     bash ./download_usps.sh   
     cd ../..
    
  4. Run experiment on MNIST.
    python DCEC.py mnist
    The DCEC model will be saved to "results/temp/dcec_model_final.h5".

  5. Run experiment on USPS.
    python DCEC.py usps

  6. Run experiment on MNIST-TEST.
    python DCEC.py mnist-test