Por último, vas a explorar visualmente los valores medios de la solución de 3 segmentos construida con NMF y podrás compararla con la de K-means. Aquí extraerás la matriz de características W, que usaremos para obtener la asignación dura de segmentos eligiendo, para cada cliente, el valor de columna (segmento) con el valor más alto asociado en esta matriz.

Hemos cargado la librería pandas como pd y la librería seaborn como sns. Se ha importado el conjunto de datos bruto wholesale, y también la instancia ya ajustada de NMF con 3 segmentos como nmf. El conjunto de datos components se ha cargado como un DataFrame de pandas.

Visualiza e interpreta las soluciones de segmentación

Create Your Free Account