用scikit-learn实现逻辑回归
时间: 2023-10-03 11:05:51 浏览: 44
scipy+scikit-learn组件
好的,下面是用scikit-learn实现逻辑回归的步骤:
1.导入所需的库
```python
from sklearn.linear_model import LogisticRegression
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score
```
2.准备数据集
```python
# 假设我们有一个二分类问题的数据集,数据集中有两个特征 X1 和 X2,以及对应的标签 y
X = [[0.3, 0.4], [0.4, 0.5], [0.5, 0.6], [0.6, 0.7], [0.7, 0.8], [0.8, 0.9]]
y = [0, 0, 0, 1, 1, 1]
```
3.划分训练集和测试集
```python
# 将数据集划分为训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
```
4.训练模型
```python
# 初始化逻辑回归模型
model = LogisticRegression()
# 训练模型
model.fit(X_train, y_train)
```
5.预测结果
```python
# 使用训练好的模型对测试集进行预测
y_pred = model.predict(X_test)
```
6.评估模型
```python
# 计算模型的准确率
accuracy = accuracy_score(y_test, y_pred)
print("Accuracy:", accuracy)
```
这样就完成了用scikit-learn实现逻辑回归的过程。
阅读全文