Source code for evalml.demos.wine

"""Load and return the wine dataset, which can be used for multiclass classification problems."""

import woodwork as ww

import evalml
from evalml.preprocessing import load_data


[docs]def load_wine(): """Load wine dataset. Multiclass problem. Returns: (pd.Dataframe, pd.Series): X and y """ filepath = ( "https://oss.alteryx.com/datasets/wine.csv?library=evalml&version=" + evalml.__version__ ) X, y = load_data(filepath, index=None, target="target") y.name = None X.ww.init() y = ww.init_series(y) return X, y