用python写一段,逻辑回归的代码
时间: 2023-02-24 08:37:26 浏览: 67
逻辑回归代码python
好的,以下是一段使用 Python 实现逻辑回归的代码:
```
import numpy as np
from sklearn.datasets import load_breast_cancer
from sklearn.linear_model import LogisticRegression
from sklearn.model_selection import train_test_split
# 加载乳腺癌数据集
data = load_breast_cancer()
X = data.data
y = data.target
# 分割数据为训练数据和测试数据
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=0)
# 训练逻辑回归模型
model = LogisticRegression()
model.fit(X_train, y_train)
# 评估模型在测试数据上的性能
accuracy = model.score(X_test, y_test)
print("Accuracy: ", accuracy)
```
这段代码先加载了一个名为 `load_breast_cancer` 的乳腺癌数据集,然后使用 `train_test_split` 函数将数据集分为训练数据和测试数据。接着使用 `LogisticRegression` 函数训练了一个逻辑回归模型,最后评估了该模型在测试数据上的性能。
阅读全文