逻辑回归-预测概率

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