【Python】逻辑回归案例代码

---

就回来吧 回来吧 有人在等你呢
有人在等你说完那句说一半的话
就别走了 留下吧 外面它太复杂
多少次让你热泪盈眶却不敢流下
铁道旁的老树下 几只乌鸦
叫到嗓音沙哑 却再没人回答
火车呼啸着驶过 驶过寂寞或繁华
曾经年轻的人啊 也会想我吗
                     🎵 毛不易《盛夏》

---

import pandas as pd
from sklearn.datasets import load_iris
from sklearn.linear_model import LogisticRegression
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score, confusion_matrix, classification_report


# 加载数据
data = load_iris()
X = data['data']
y = data['target']

# 将数据分为训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=1)

# 将测试集转换为DataFrame以便更好地展示
test_df = pd.DataFrame(X_test, columns=data['feature_names'])
test_df['target'] = y_test

# 显示测试集的前五个样本
print(test_df.head())

model = LogisticRegression(max_iter=200)

# 训练模型
model.fit(X_train, y_train)
# 在测试集上进行预测
y_pred = model.predict(X_test)

# 评估模型
print("Accuracy:", accuracy_score(y_test, y_pred))
print("Confusion Matrix:")
print(confusion_matrix(y_test, y_pred))
print("Classification Report:")
print(classification_report(y_test, y_pred))