逻辑回归lr机器学习算法
时间: 2024-01-16 21:17:36 浏览: 88
机器学习 LR逻辑回归算法
逻辑回归(Logistic Regression)是一种常用的机器学习算法,用于处理分类问题。它可以用于二分类问题或多分类问题。在scikit-learn库中,提供了一个LogisticRegression类来实现逻辑回归模型。
逻辑回归的算法原理是基于线性回归的思想,通过将线性回归的结果映射到一个概率值,然后根据概率值进行分类。逻辑回归使用的是sigmoid函数(也称为逻辑函数)来将线性回归的结果映射到0到1之间的概率值。当概率值大于等于0.5时,将样本分类为正类;当概率值小于0.5时,将样本分类为负类。
下面是一个使用scikit-learn中的LogisticRegression类进行逻辑回归的示例代码:
```python
from sklearn.linear_model import LogisticRegression
# 创建逻辑回归模型对象
lr = LogisticRegression()
# 准备训练数据和标签
X_train = [[1, 2], [3, 4], [5, 6]]
y_train = [0, 1, 0]
# 训练模型
lr.fit(X_train, y_train)
# 准备测试数据
X_test = [[2, 3], [4, 5]]
# 预测测试数据的分类
y_pred = lr.predict(X_test)
# 打印预测结果
print("预测结果:", y_pred)
```
这段代码中,首先导入了LogisticRegression类,然后创建了一个逻辑回归模型对象lr。接着准备了训练数据X_train和对应的标签y_train,使用fit()方法对模型进行训练。然后准备了测试数据X_test,使用predict()方法对测试数据进行分类预测,最后打印出预测结果。
阅读全文