在上一篇博客中我们学习了逻辑回归(LogisticRegression)的理论。那么在这篇博客中,我们用代码展示一下,如何用梯度下降法获取逻辑回归的参数
步骤1:我们加载sklearn中的鸢尾花数据进行测试,由于为了数据可视化,我们选择2种类型的鸢尾花,并且只选择2个特征。
import numpy as np
import matplotlib.pyplot as plt
from sklearn import datasets
X, y = datasets.load_iris(return_X_y=True)
X = X[y < 2, :2]
y = y[y < 2]
plt.scatter(X[y == 0, 0], X[y == 0, 1], color="red")
plt.scatter(X[y == 1, 0], X[y == 1, 1], color="blue")
plt.show()
可视化一下:
步骤二: 我们编写自己的回归算法