决策树-Heart分类

def solve():
    import pandas as pd
    from sklearn.tree import DecisionTreeClassifier
    from sklearn.model_selection import train_test_split
    from sklearn.metrics import accuracy_score, recall_score
    df = pd.read_csv('https://liangdaima.com/static/data/heart.csv')
    X = df.drop('target', axis=1)
    y = df['target']
    X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
    dt = DecisionTreeClassifier(random_state=42)
    dt.fit(X_train, y_train)
    y_pred = dt.predict(X_test)
    print('准确率:', round(accuracy_score(y_test, y_pred), 4))
    print('召回率:', round(recall_score(y_test, y_pred), 4))