A pytorch implementation of the paper Unsupervised Deep Embedding for Clustering Analysis.
Clone the project into your local system
git clone https://github.com/Deepayan137/DeepClustering.git
cd DeepClustering
- python 3.5+
- pytorch 1.0
conda install --file requirements.txt
python -m main --train_epochs 200 --pretrain_epochs 100