discrimintools.datasets.load_heart

discrimintools.datasets.load_heart(element='subset')

Heart dataset

Parameters:

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

• ‘subset’ for subset of all dataset (with few columns).

• ‘train’ for training dataset.

• ‘test’ for testing dataset.

Returns:

heart – The heart 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.datasets import load_heart
>>> from discrimintools import GFALDA
>>> D = load_heart("subset") # load subset data
>>> y, X = D["disease"], D.drop(columns=["disease"]) # split into X and y
>>> clf = GFALDA(n_components=5)
>>> clf.fit(X,y)
GFALDA(n_components=5)