决策树-预测概率

def solve():
    import pandas as pd
    import numpy as np
    from sklearn.tree import DecisionTreeClassifier
    df = pd.read_csv('https://liangdaima.com/static/data/iris.csv')
    X = df.drop('species', axis=1)
    y = df['species']
    dt = DecisionTreeClassifier(random_state=42)
    dt.fit(X, y)
    sample = np.array([[5.1, 3.5, 1.4, 0.2]])
    proba = dt.predict_proba(sample)
    for i, cls in enumerate(dt.classes_):
        print(f'{cls}: {proba[0][i]:.4f}')