Mélange de gaussiennes

import seaborn as sns

import pandas as pd

iris = sns.load_dataset("iris")

X = iris.iloc[:,:4]

y = iris.species

from sklearn.mixture import GaussianMixture

# instanciation

gm = GaussianMixture( n_components=3)

# ajustement

gm.fit(X)

# prediction

gm.predict(X)

#

gm.predict_proba(X) # La probabilité la plus forte compte

import matplotlib.pyplot as plt

plt.scatter(X.iloc[:,0],X.iloc[:,1],c=gm.predict(X))