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