📜  Python教程中的逻辑回归(1)

📅  最后修改于: 2023-12-03 15:04:39.489000             🧑  作者: Mango

Python教程中的逻辑回归

逻辑回归是机器学习中的一种分类算法。它被广泛应用于许多领域,例如金融、医疗和社交网络分析等。在Python中,我们可以使用Scikit-learn库来构建和拟合逻辑回归模型。

数据集

在介绍逻辑回归算法之前,我们需要一个数据集。在这个例子中,我们将使用鸢尾花数据集。这个数据集有150个样本,分为三类:Setosa,Versicolor和Virginica。每个样本有四个特征:花萼长度、花萼宽度、花瓣长度、花瓣宽度。我们的任务是根据这四个特征预测鸢尾花的类别。

from sklearn.datasets import load_iris

iris = load_iris()

X = iris.data
y = iris.target
数据预处理

在训练之前,我们需要对数据进行一些预处理。在这个例子中,我们将把数据集分成训练集和测试集。训练集用于训练模型,测试集用于评估模型的准确性。我们将使用Scikit-learn库中的train_test_split()函数来完成此操作。

from sklearn.model_selection import train_test_split

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
构建模型

现在我们将使用Scikit-learn库中的LogisticRegression()函数来构建模型。我们将训练模型,并使用测试集来评估它的准确性。

from sklearn.linear_model import LogisticRegression

model = LogisticRegression()
model.fit(X_train, y_train)

score = model.score(X_test, y_test)
print("Model Accuracy: {:.2f}%".format(score * 100))

结果:

Model Accuracy: 100.00%

在这个例子中,我们得到了100%的模型准确度。这是因为鸢尾花数据集是一个相对简单的问题,并且Logistic回归是一个相对简单而又强大的算法。但是在实际问题中,你可能需要使用更复杂的算法来解决问题。

结论

逻辑回归是一种强大而灵活的算法,可以用于分类问题。在Python中,我们可以使用Scikit-learn库来构建和拟合逻辑回归模型。在我们的例子中,我们使用了鸢尾花数据集,并得出了100%的模型准确率。