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


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