机器学习算法:从头开始构建逻辑回归模型

原作:Rohith Gandhi

郭一璞 编译自 Hacher Noon

量子位 出品 | 公众号 QbitAI

逻辑回归是继线性回归之后最著名的机器学习算法。

在很多方面,线性回归和逻辑回归是相似的,不过最大的区别在于它们的用途,线性回归算法用于预测,但逻辑回归用于分类任务。

分类任务很常见,比如把电子邮件分为垃圾邮件和非垃圾邮件、把肿瘤分为恶性或者良性、把网站分为危险站点或正常站点,机器学习算法就可以完成这些任务。

其中,逻辑回归算法就是一种分类算法,简单粗暴,但有用。

现在,开始深入研究逻辑回归。

Sigmoid函数(Logistic函数)

逻辑回归算法使用具有独立预测因子的线性方程来预测,预测值可以是从负无穷到正无穷之间的任何值。

我们需要让算法的输出为类变量,比如用0表示非,用1表示是。

因此,我们将线性方程的输出压缩到[0,1]的范围内。

为了压缩0和1之间的预测值,我们使用sigmoid函数:

线性方程和sigmoid函数

压缩输出-h

我们取线性方程的输出(z)并给出返回压缩值h的函数g(x),h将位于0到1的范围内。为了理解sigmoid函数如何压缩,我们画出了sigmoid函数的图形:

sigmoid函数图形

如图可见,sigmoid函数当x>0时,y逐渐向1靠近;当x

成本函数(cost function)

由于我们试图预测类别值,不能使用和线性回归算法中相同的成本函数。

所以,我们使用损失函数的对数来计算错误分类的成本。

考虑到计算上面这个函数的梯度实在太难了,,我们把它写成下面这个样子:

计算梯度

我们取相对于每个参数(θ_0,θ_1,…)的成本函数的偏导数来获得梯度,有了这些梯度,我们可以更新θ_0,θ_1,…的值。

现在,开始召唤微积分大法:

梯度

如果看懂了,那你的微积分学得棒棒的。

不过,如果微积分大法召唤失败……就直接照着上面的式子做吧。

写代码

现在方程终于搞定了,开始写代码。

我们用NumPy来从头构建模型,用IRIS(鸢尾花)数据集来训练和测试算法。

1import pandas as pd

2

3df = pd.read_csv('/Users/rohith/Documents/Datasets/Iris_dataset/iris.csv') ## Load data

4df = df.drop(['Id'],axis=1)

5rows = list(range(100,150))

6df = df.drop(df.index[rows]) ## Drop the rows with target values Iris-virginica

7Y = []

8target = df['Species']

9for val in target:

10 if(val == 'Iris-setosa'):

11 Y.append(0)

12 else:

13 Y.append(1)

14df = df.drop(['Species'],axis=1)

15X = df.values.tolist()

我们用pandas来加载数据。

IRIS数据集有三个目标值,分别是弗吉尼亚鸢尾、山鸢尾、变色鸢尾。但是因为要实现的是二进制的分类算法,所以此处先把弗吉尼亚鸢尾剔除。

变色鸢尾(左)和山鸢尾(右),图源百度百科

现在,只剩下两个目标值用来分类了。

之后,从数据集中提取独立变量和因变量,现在可以继续准备训练集和测试集了。

1from sklearn.utils import shuffle

2from sklearn.cross_validation import train_test_split

3import numpy as np

4

5X, Y = shuffle(X,Y)

6

7x_train = []

8y_train = []

9x_test = []

10y_test = []

11

12x_train, x_test, y_train, y_test = train_test_split(X, Y, train_size=0.9)

13

14x_train = np.array(x_train)

15y_train = np.array(y_train)

16x_test = np.array(x_test)

17y_test = np.array(y_test)

18

19x_1 = x_train[:,0]

20x_2 = x_train[:,1]

21x_3 = x_train[:,2]

22x_4 = x_train[:,3]

23

24x_1 = np.array(x_1)

25x_2 = np.array(x_2)

26x_3 = np.array(x_3)

27x_4 = np.array(x_4)

28

29x_1 = x_1.reshape(90,1)

30x_2 = x_2.reshape(90,1)

31x_3 = x_3.reshape(90,1)

32x_4 = x_4.reshape(90,1)

33

34y_train = y_train.reshape(90,1)

我们清洗了数据,并且把它们分为了训练集和测试集,训练集中有90个数据,测试集中有10个数据。由于数据集中有四个预测因子,所以我们提取每个特征并将其存储在各个向量中。

1## Logistic Regression

2import numpy as np

3

4def sigmoid(x):

5 return (1 / (1 + np.exp(-x)))

6

7m = 90

8alpha = 0.0001

9

10theta_0 = np.zeros((m,1))

11theta_1 = np.zeros((m,1))

12theta_2 = np.zeros((m,1))

13theta_3 = np.zeros((m,1))

14theta_4 = np.zeros((m,1))

15

16

17epochs = 0

18cost_func = []

19while(epochs

20 y = theta_0 + theta_1 * x_1 + theta_2 * x_2 + theta_3 * x_3 + theta_4 * x_4

21 y = sigmoid(y)

22

23 cost = (- np.dot(np.transpose(y_train),np.log(y)) - np.dot(np.transpose(1-y_train),np.log(1-y)))/m

24

25 theta_0_grad = np.dot(np.ones((1,m)),y-y_train)/m

26 theta_1_grad = np.dot(np.transpose(x_1),y-y_train)/m

27 theta_2_grad = np.dot(np.transpose(x_2),y-y_train)/m

28 theta_3_grad = np.dot(np.transpose(x_3),y-y_train)/m

29 theta_4_grad = np.dot(np.transpose(x_4),y-y_train)/m

30

31 theta_0 = theta_0 - alpha * theta_0_grad

32 theta_1 = theta_1 - alpha * theta_1_grad

33 theta_2 = theta_2 - alpha * theta_2_grad

34 theta_3 = theta_3 - alpha * theta_3_grad

35 theta_4 = theta_4 - alpha * theta_4_grad

36

37 cost_func.append(cost)

38 epochs += 1

我们用0来初始化参数(θ_0,θ_1,…)。当我们使用线性方程来计算这些值时,这些值将被压缩到0到1的范围内。

然后计算成本。

可以用成本函数计算每个参数的梯度,并通过将梯度与α相乘来更新它们的值,α是算法的学习率。一万次之后,我们的算法会收敛到最小值。

现在,终于可以找出那10个测试集的数据,开始测试了。

1from sklearn.metrics import accuracy_score

2

3test_x_1 = x_test[:,0]

4test_x_2 = x_test[:,1]

5test_x_3 = x_test[:,2]

6test_x_4 = x_test[:,3]

7

8test_x_1 = np.array(test_x_1)

9test_x_2 = np.array(test_x_2)

10test_x_3 = np.array(test_x_3)

11test_x_4 = np.array(test_x_4)

12

13test_x_1 = test_x_1.reshape(10,1)

14test_x_2 = test_x_2.reshape(10,1)

15test_x_3 = test_x_3.reshape(10,1)

16test_x_4 = test_x_4.reshape(10,1)

17

18index = list(range(10,90))

19

20theta_0 = np.delete(theta_0, index)

21theta_1 = np.delete(theta_1, index)

22theta_2 = np.delete(theta_2, index)

23theta_3 = np.delete(theta_3, index)

24theta_4 = np.delete(theta_4, index)

25

26theta_0 = theta_0.reshape(10,1)

27theta_1 = theta_1.reshape(10,1)

28theta_2 = theta_2.reshape(10,1)

29theta_3 = theta_3.reshape(10,1)

30theta_4 = theta_4.reshape(10,1)

31

32y_pred = theta_0 + theta_1 * test_x_1 + theta_2 * test_x_2 + theta_3 * test_x_3 + theta_4 * test_x_4

33y_pred = sigmoid(y_pred)

34

35new_y_pred =[]

36for val in y_pred:

37 if(val >= 0.5):

38 new_y_pred.append(1)

39 else:

40 new_y_pred.append(0)

41

42print(accuracy_score(y_test,new_y_pred))

提前准备好的测试集和训练集的特征十分相似,但是因为测试示例的数量只有10个,所以我们把θ_0,θ_1,θ_2,θ_3和θ_4的值从90×1剪切到10×1,计算了测试类别并检查模型的精确度。

哎呦不错

完美!模型准确度100%!

虽然逻辑回归算法非常强大,但我们使用的数据集并不是很复杂,所以我们的模型能够达到100%的准确度。

我们还可以画出一万次训练的成本函数图:

1import matplotlib.pyplot as plt

2

3cost_func = np.array(cost_func)

4cost_func = cost_func.reshape(10000,1)

5plt.plot(range(len(cost_func)),cost_func)

您的成本函数请查收

不过,现在你可能觉得这个算法的代码太多了。为了缩短代码行数,我们用上了scikit学习库。scikit学习库有一个内置的逻辑回归类别,我们可以直接接入使用。

1from sklearn.metrics import accuracy_score

2from sklearn.linear_model import LogisticRegression

3

4clf = LogisticRegression()

5clf.fit(x_train,y_train)

6y_pred = clf.predict(x_test)

7print(accuracy_score(y_test,y_pred))

看,代码已经被缩减到10行以内了。用scikit学习库,我们的模型准确率依然100%。

https://hackernoon.com/introduction-to-machine-learning-algorithms-logistic-regression-cbdd82d81a36

  • 发表于:
  • 原文链接https://kuaibao.qq.com/s/20180530A090UR00?refer=cp_1026
  • 腾讯「云+社区」是腾讯内容开放平台帐号(企鹅号)传播渠道之一,根据《腾讯内容开放平台服务协议》转载发布内容。
  • 如有侵权,请联系 yunjia_community@tencent.com 删除。

同媒体快讯

扫码关注云+社区

领取腾讯云代金券