Clustering avec KMeans
L'algorithme des K Moyennes communément appelé KMeans est un algorithme de clustering.
L'algorithme du KMeans
Les pre-réquis
Pour utiliser l'algorithme du KMeans pour former un modèle, il faut des données numériques.
Fonction de coût
L'écart
Méthode d'optimisation
Condition d'arrêt
Évaluation
Pour évaluer un modèle de KMeans, on peut s'intéresser à : - la forme des clusters - la stabilité de l'algorithme - la compatibilté des résuktats avec des
.
Module 3: Apprentissage non supervisé
Partie 3.2: Clustering avec KMEANS
Il s'agit d'une technique linéaire permettant de projeter des données X de dimenson p en d'autres données T de dimmension d < p.
Objectifs * Visualiser les données * Étudier les liens de corrélation linéaire entre les variables * Réduire le nombre de dimensions
L'idée de l'ACP est de projeter les données sur des axes préservant la variance des données. Avec p=2 ou p=3, on peut visualiser ls données.
Principe : approximer un nuage de points X de dimension p par sa projection linéaire en dimension d < p.
L'ACP maximise la variance des points projectés.
On cherche à réaliser la projection x app Rp-->Rd, d<p
L'espace de projection P inclus dans Rd sera construit de manière progressive.
* D'abord on va chercher le meilleur axe de projection (1D) u1
* Ensuite le meilleur plan en trouvant le deuxième axe u2
* Et ainsi de suite jusqu'à obtenir P
Importation des données
Le jeu de donnés contient des donnés récuellies sur les pays pour l'année 2015, il comporte 158 lignes et 12 colonnes. Nous travaillerons avec les 8 dernières colonnes soit p=8 et la colonne Region pour évaluer notre ACP.
fig = plt.figure(figsize=(16, 8))
gs = fig.add_gridspec(2, 2)
fig.add_subplot(gs[0, 0])
plt.scatter(X[:, 1], X[:, 4])
fig.add_subplot(gs[0, 1])
plt.scatter(X[:, 10], X[:, 25])
fig.add_subplot(gs[1, 0])
plt.scatter(X[:, 7], X[:, 14])
fig.add_subplot(gs[1, 1])
plt.scatter(X[:, 6], X[:, 40])
plt.show()
KMEANS avec Scikit-learn
Comment choisir le nombre de clusters ?
from sklearn.metrics import silhouette_score
import matplotlib.pyplot as plt
n_clusters_range = [2, 3, 4, 5, 6, 7]
silouettes = []
for n in n_clusters_range:
kmeans = KMeans(n_clusters=n)
kmeans.fit(X)
silouettes.append(silhouette_score(X, kmeans.predict(X)))
plt.plot(n_clusters_range, silouettes)
plt.xlabel("Nombre de clusters")
plt.ylabel("Coefficient de silhouette")
plt.title("Coefficient de silhouette en fonction du nombre de clusters")
plt.show()
fig = plt.figure(figsize=(16, 8))
gs = fig.add_gridspec(2, 2)
fig.add_subplot(gs[0, 0])
plt.scatter(X[:, 1], X[:, 4], c=kmeans.predict(X))
fig.add_subplot(gs[0, 1])
plt.scatter(X[:, 10], X[:, 25], c=kmeans.predict(X))
fig.add_subplot(gs[1, 0])
plt.scatter(X[:, 7], X[:, 14], c=kmeans.predict(X))
fig.add_subplot(gs[1, 1])
plt.scatter(X[:, 6], X[:, 40], c=kmeans.predict(X))
plt.show()
Interprétaion
Le premier axe conserve 41,2% le deuxième axe 19,4% et le troisème axe 14,3%. Les trois axes permettent de conserver environ 75% de variance des données. Si l'on souhaite conserver plus de variance, on peut ajouter un axe supplémentaire ou plus. Remarquons qu'il est peu pertinent de conserver toute la variance, autant travailler avec les données initiales, il ne faut pas perdre de vue que l'un des objectifs de l'ACP, c'est la réduction du nombre de dimensions.