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

Cluster Visualization