Skip to content

sohailkhanmarwat/Deep-Clustering

 
 

Repository files navigation

Deep-Clustering

A Convolutional Neural Network based model for Unsupervised Learning

Results

Here is a comparison plot of K-Means and our CNN based model on 2D data generated from two Gaussian samples
      Input Plot               K-Means Plot             NN Plot

We next performed Clustering on MNIST dataset(784 dimensional data) and extracted a few samples from each cluster
Our CNN based model far exceeded the K-Means algorithm in NMI score

About

A Convolutional Neural Network based model for Unsupervised Learning

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%