"""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://api.featurelabs.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