决策树过拟合可视化

def solve():
    import pandas as pd
    import matplotlib.pyplot as plt
    from sklearn.tree import DecisionTreeClassifier
    from sklearn.model_selection import train_test_split
    df = pd.read_csv(____)
    X = df.drop(____); y = df["____"]
    X_train, X_test, y_train, y_test = train_test_split(____)
    depths = [____, ____, ____, ____, None]
    tr, te = [], []
    for d in depths:
        tree = DecisionTreeClassifier(____).fit(____)
        tr.append(tree.score(____)); te.append(tree.score(____))
    labels = [str(____) for d in depths]
    plt.plot(____); plt.plot(____)
    plt.xlabel(____); plt.ylabel(____); plt.legend()
    plt.title(____); plt.show()
    print(____)