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