Source code for evalml.demos.breast_cancer

import pandas as pd
import woodwork as ww
from sklearn.datasets import load_breast_cancer as load_breast_cancer_sk


def load_breast_cancer():
    """Load breast cancer dataset. Binary classification problem.

    Returns:
        (pd.Dataframe, pd.Series): X and y
    """
    data = load_breast_cancer_sk()
    X = pd.DataFrame(data.data, columns=data.feature_names)
    y = pd.Series(data.target)
    y = y.map(lambda x: data["target_names"][x])

    X.ww.init()
    y = ww.init_series(y)
    return X, y