Implementation of K-means Clustering

[ckeshava]

Description

Implement the k-means clustering algorithm.

Details

Technical Specifications: Tensorflow/PyToch/Caffe

• Type of issue: Multiple (Implementations in different frameworks will be considered separately)
• Time Limit: 3 days

Issue requirements / progress

• Tensorflow: Provide examples that show that the results are dependent on the initial - choice of the centroids

• Tensorflow: Plot the elbow curve after running the algorithm for different values of centers, and suggest an optimal value for the same.

• Pytorch: Provide examples that show that the results are dependent on the initial - choice of the centroids

• Pytorch: Plot the elbow curve after running the algorithm for different values of centers, and suggest an optimal value for the same.

Directory Structure

Name the code file as <your_name>.py and save it inside this folder. Assuming this is implemented in tensorflow: machine_learning/kmeans_clustering/tensorflow/your_implementation.py/ck.py

Note

Please claim the issue first by commenting here before starting to work on it.

[mehnazyunus]

Hi, I'd like to work on the Tensorflow implementation.

[mahim23]

@mehnazyunus sure go ahead!

[GajeshS]

Hi, I would like to work on the Pytorch implementation of K-Means Clustering.