discrimintools.datasets.load_mushroom#

discrimintools.datasets.load_mushroom(element='train')[source]#

Mushroom dataset

Parameters:

element (str, default = ‘train’) – The dataset to load. Possible values are:

  • ‘train’ for training dataset.

  • ‘test’ for testing dataset.

Returns:

mushroom – The mushroom dataset.

Return type:

DataFrame of shape (n_samples, n_columns)

References

[1] Ricco Rakotomalala (2020), « Pratique de l’Analyse Discriminante Linéaire », Version 1.0, Université Lumière Lyon 2.

Examples

>>> from discrimintools.dataset import load_mushroom
>>> from discrimintools import GFALDA
>>> D = load_mushroom("train") # load training data
>>> y, X = D["classe"], D.drop(columns=["classe"]) # split into X and y
>>> clf = GFALDA(n_components=5)
>>> clf.fit(X,y)
GFALDA(n_components=5)