Mini-Batch K-Means

Clustering DS practice problem on Onlearn.

Difficulty: medium.

Topics: Understanding Mini-Batch K-Means Optimization, Mini-batch size selection, Update step for centroids, Random state reproducibility, Assignment step complexity, Memory efficiency in clustering, Unsupervised Learning, Clustering Algorithms, Optimization Techniques, Computational Complexity, Iterative Refinement, Centroid Initialization, Batch Processing, Stochastic Gradient Descent, Convergence Criteria, Inertia Calculation.

Implement a function that performs Mini Batch K Means clustering on a given dataset X with a specified number of clusters k. Ensure the function returns the cluster labels for the input data.