KMeans Clustering Implementation with TensorFlow and Performance Comparison with SkLearn KMeans - Deep Cognition Labs

This post describes implementation of K-Means Clustering algorithm using TensorFlow. I have tested the code with GPU (Nvidia GTX 1080 Founders Edition) accelerated TensorFlow and for large dataset it seems to be 2-3 times faster than the CPU based sklearn Kmeans implementation based on number of samples.