Exercise

Run k-means

You will now build a 3 clusters with k-means clustering. We have loaded the pre-processed RFM dataset as datamart_normalized. We have also loaded the pandas library as pd.

You can explore the dataset in the console to get familiar with it.

Instructions

100 XP

Import KMeans from the scikit-learn library.
Initialize KMeans with 3 clusters and random state 1.
Fit k-means clustering on the normalized data set.
Extract cluster labels and store them as cluster_labels.