def k_means(data_points, n_centroids, tot_iteration=100):
centroids = kmeans.create_centroids(data_points, n_centroids)
[cluster_label,
new_centroids] = kmeans.iterate_k_means(data_points, centroids,
tot_iteration)
# kmeans.print_label_data([cluster_label, new_centroids])
return [cluster_label, new_centroids]
