/DeepClustering

A pytorch implementation of the paper Unsupervised Deep Embedding for Clustering Analysis.

Primary LanguagePython

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

Results

Cluster Visualization