c语言编程笔录

首页 >   > 笔记大全

笔记大全

Python中的逻辑回归实例

更新时间:2023-08-15

背景介绍

逻辑回归是一种常用的分类算法,可以用来解决二分类问题。它基于统计学原理,将输入特征与输出结果之间的关系建模为一个概率函数,并通过最大似然估计来寻找最优参数。

代码实现

import numpy as np
from sklearn.linear_model import LogisticRegression

# 创建训练数据
X_train = np.array([[1, 2], [2, 4], [3, 6], [4, 8], [5, 10]])
y_train = np.array([0, 0, 0, 1, 1])

# 创建逻辑回归模型并进行训练
model = LogisticRegression()
model.fit(X_train, y_train)

# 预测新样本
X_test = np.array([[6, 12], [7, 14]])
y_pred = model.predict(X_test)

代码解释

以上代码首先导入了必要的库,然后创建了训练数据集,其中X_train是输入特征,y_train是对应的分类标签。接下来,通过调用sklearn库中的LogisticRegression类,我们创建了一个逻辑回归模型,并使用fit方法对模型进行训练。

最后,我们使用X_test作为输入特征,调用predict方法预测其分类标签,并将结果存储在y_pred中。

总结

通过使用Python中的逻辑回归算法,我们可以轻松实现二分类任务。该算法基于统计学原理,适用于许多实际应用场景。希望以上内容对你理解和应用逻辑回归提供了一些帮助。