Clustering con K-Means

K-Means es uno de los algoritmos de clustering más populares. Divide los datos en una cantidad predefinida de clústeres (K) minimizando las distancias intraclúster.

Funciona de forma iterativa para encontrar las mejores asignaciones de clústeres y centroides.

Cómo funciona K-Means

K-Means funciona siguiendo estos pasos:

1. Inicializar K centroides aleatorios.
2. Asignar cada punto de datos al centroide más cercano.
3. Recalcular los centroides como la media de todos los puntos de un clúster.
4. Repetir los pasos 2 y 3 hasta que los centroides se estabilicen o se alcance el número máximo de iteraciones.

Elección del número de clústeres

El valor de K es un parámetro fundamental. Entre los métodos habituales para elegir K se incluyen:

• Método del codo: Representar la suma de cuadrados intra-clúster (WCSS) para distintos valores de K y buscar el «punto de codo».
• Índice de silueta: Mide el grado de separación entre los clústeres.

Conjunto de datos de ejemplo para K-Means

Consideremos un conjunto de datos con puntos distribuidos en un espacio 2D. Cada punto tiene:

• X: Coordenada en el eje horizontal.
• Y: Coordenada en el eje vertical.

K-Means agrupará estos puntos en clústeres según su proximidad.

Implementar K-Means en Python

Utilizaremos la clase KMeans de la biblioteca scikit-learn para implementar el algoritmo:

from sklearn.cluster import KMeans
import numpy as np

# Example dataset
X = np.array([[1, 2], [1, 4], [1, 0], [4, 2], [4, 4], [4, 0]])

# K-Means model
kmeans = KMeans(n_clusters=2, random_state=0)
kmeans.fit(X)

print("Cluster Centers:", kmeans.cluster_centers_)
print("Labels:", kmeans.labels_)

Visualizar clústeres

Podemos visualizar los clústeres representando gráficamente los puntos de datos y los centroides:

import matplotlib.pyplot as plt

# Scatter plot of data points
plt.scatter(X[:, 0], X[:, 1], c=kmeans.labels_, cmap='viridis')

# Plot cluster centers
plt.scatter(kmeans.cluster_centers_[:, 0], kmeans.cluster_centers_[:, 1], s=200, c='red', marker='X')
plt.title('K-Means Clustering')
plt.show()

Ventajas de K-Means

K-Means ofrece varias ventajas:

• Es sencillo y fácil de implementar.
• Es eficiente con conjuntos de datos grandes.
• Funciona bien cuando los clústeres están claramente definidos.

Limitaciones de K-Means

Algunas limitaciones de K-Means son:

• Su sensibilidad a la posición inicial de los centroides.
• La dificultad para manejar clústeres no esféricos.
• Su sensibilidad a los valores atípicos y a los datos ruidosos.

Resumen y próximos pasos

En esta lección, hemos visto:

• Cómo funciona el clustering de K-Means.
• Cómo elegir el número de clústeres mediante métodos como el método del codo.
• Cómo implementar y visualizar K-Means en Python.

A continuación, aplicaremos K-Means a un conjunto de datos del mundo real para adquirir experiencia práctica.