1. Learn
    2. /
  2. Courses
    3. /
  3. Customer Segmentation in Python

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.