modal-inria / MixtComp

Model-based clustering package for mixed data
Other
12 stars 4 forks source link

basic_mode modifies the user's data set #16

Open Quentin62 opened 2 years ago

Quentin62 commented 2 years ago
from pyMixtComp import MixtComp
from pyMixtComp.data import load_iris

iris, _ = load_iris()
iris = iris.rename(columns={"species": "z_class"})
iris
sepal length (cm) sepal width (cm) petal length (cm) petal width (cm) species
5.1 3.5 1.4 0.2 setosa
4.9 3.0 1.4 0.2 setosa
4.7 3.2 1.3 0.2 setosa
4.6 3.1 1.5 0.2 setosa
5.0 3.6 1.4 0.2 setosa
... ... ... ... ...
6.7 3.0 5.2 2.3 virginica
6.3 2.5 5.0 1.9 virginica
6.5 3.0 5.2 2.0 virginica
6.2 3.4 5.4 2.3 virginica
5.9 3.0 5.1 1.8 virginica
mod = MixtComp(n_components=3, n_init=5)
mod.fit(iris)
iris
  sepal length (cm) sepal width (cm) petal length (cm) petal width (cm) z_class
5.1 3.5 1.4 0.2 0
4.9 3.0 1.4 0.2 0
4.7 3.2 1.3 0.2 0
4.6 3.1 1.5 0.2 0
5.0 3.6 1.4 0.2 0
... ... ... ... ...
6.7 3.0 5.2 2.3 2
6.3 2.5 5.0 1.9 2
6.5 3.0 5.2 2.0 2
6.2 3.4 5.4 2.3 2
5.9 3.0 5.1 1.8 2