逻辑回归sklearn
时间: 2023-08-07 15:06:11 浏览: 122
python sklearn库实现简单逻辑回归的实例代码
3星 · 编辑精心推荐
逻辑回归是一种用于二分类问题的机器学习算法。在Scikit-learn库中,可以使用LogisticRegression类来实现逻辑回归。
首先,你需要导入LogisticRegression类:
```python
from sklearn.linear_model import LogisticRegression
```
然后,你可以创建一个LogisticRegression对象,并设置相关的参数:
```python
model = LogisticRegression()
```
接下来,你需要准备你的训练数据和标签。训练数据是一个二维数组,每一行代表一个样本,每一列代表一个特征。标签是一个一维数组,用于表示每个样本的类别。
然后,你可以使用fit方法来训练模型:
```python
model.fit(X_train, y_train)
```
其中,X_train是训练数据,y_train是对应的标签。
训练完成后,你可以使用predict方法来预测新的样本的类别:
```python
y_pred = model.predict(X_test)
```
其中,X_test是测试数据,y_pred是预测结果。
你也可以使用score方法来评估模型在测试数据上的准确率:
```python
accuracy = model.score(X_test, y_test)
```
其中,X_test是测试数据,y_test是对应的标签。
这就是使用Scikit-learn库中的LogisticRegression类实现逻辑回归的基本步骤。当然,你还可以通过设置不同的参数来进行模型调优,例如正则化参数C的值等。
阅读全文