本文由機器之心編輯,「機器之心」專注生產人工智慧專業性內容,適合開發者和從業者閱讀參考。點擊右上角即刻關注。
logistic 回歸是一種著名的二元分類問題的線性分類算法。它容易實現、易於理解,並在各類問題上有不錯的效果,即使該方法的原假設與數據有違背時。
在本教程中,你將了解如何在 Python 中實現隨機梯度下降的 logistic 回歸算法。學完本教程後,你將了解:
如何使用 logistic 回歸模型進行預測。
如何使用隨機梯度下降(stochastic gradient descent)來估計係數(coefficient)。
如何將 logistic 回歸應用到真實的預測問題。
讓我們開始吧!
描述
本節將簡要介紹 logistic 回歸算法、隨機梯度下降以及本教程使用的 Pima 印第安人糖尿病數據集。
logistic 回歸算法
logistic 回歸算法以該方法的核心函數命名,即 logistic 函數。logistic 回歸的表達式為方程,非常像線性回歸。輸入值(X)通過線性地組合權重或係數值來預測輸出值(y)。
與線性回歸的主要區別在於,模型的輸出值是二值(0 或 1),而不是連續的數值。
yhat = e^(b0 + b1 * x1) / (1 + e^(b0 + b1 * x1))
公式可簡化為:
yhat = 1.0 / (1.0 + e^(-(b0 + b1 * x1)))
這裡 e 是自然對數的底(歐拉數),yhat 是預測值,b0 是偏差或截距項,b1 是單一輸入變量(x1)的參數。
yhat 預測值為 0 到 1 之間的實數,它需要捨入到整數值並映射到預測類值。
輸入數據中的每一列都有一個相關係數 b(一個常數實數值),這個係數是從訓練集中學習的。存儲在存儲器或文件中的最終模型的實際上是等式中的係數(β值或 b)。
logistic 回歸算法的係數必須從訓練集中估計。
隨機梯度下降
梯度下降是通過順著成本函數(cost function)的梯度來最小化函數的過程。
這涉及到成本函數的形式及導數,使得從任意給定點能推算梯度並在該方向上移動,例如,沿坡向下(downhill)直到最小值。
在機器學習中,我們可以使用一種技術來評估和更新每次迭代後的係數,這種技術稱為隨機梯度下降,它可以使模型的訓練誤差(training error)最小化。
此優化算法每次將每個訓練樣本傳入模型。模型對訓練樣本進行預測,計算誤差並更新模型以便減少下一預測的誤差。
該過程可以找到使訓練誤差最小的一組係數。每次迭代,機器學習中的係數(b)通過以下等式更新:
b = b + learning_rate * (y - yhat) * yhat * (1 - yhat) * x
其中 b 是被優化的係數或權重,learning_rate 是必須設置的學習速率(例如 0.01),(y - y hat)是訓練數據基於權重計算的模型預測誤差,y hat 是通過係數計算的預測值,x 是輸入值
Pima 印第安人糖尿病數據集
Pima Indians 數據集包含了根據基本醫療細節預測 Pima 印第安人 5 年內糖尿病的發病情況。
它是一個二元分類問題,其中預測是 0(無糖尿病)或 1(糖尿病)。
它包含 768 行和 9 列。所有值都是數字型數值,含有浮點值(float)。下面的例子展示了數據前幾行的結構。
6,148,72,35,0,33.6,0.627,50,1
1,85,66,29,0,26.6,0.351,31,0
8,183,64,0,0,23.3,0.672,32,1
1,89,66,23,94,28.1,0.167,21,0
0,137,40,35,168,43.1,2.288,33,1
...
通過預測多數類(零規則算法 Zero Rule Algorithm),這個問題的基線性能為 65.098%的分類準確率(accuracy)。你可以在 UCI 機器學習資料庫中了解有關此數據集的更多信息:https://archive.ics.uci.edu/ml/datasets/Pima+Indians+Diabetes
下載數據集,並將其保存到你當前的工作目錄,文件名為 pima-indians-diabetes.csv。
教程
本教程分為 3 部分。
1. 進行預測
2. 估計係數
3. 糖尿病數據集預測
學完這三部分,你將具有應用 logistic 回歸與隨機梯度下降的基礎,並可以開始處理你自己的預測建模問題。
1. 進行預測
第一步是開發一個可以進行預測的函數。
在隨機梯度下降中估計係數值以及模型最終確定後在測試集上進行預測都需要這個預測函數。
下面是一個名為 predict() 的函數,給定一組係數,它預測每一行的輸出值。
第一個係數始終為截距項 (intercept),也稱為偏差或 b0,因為它是獨立的,不是輸入值的係數。
# Make a prediction with coefficients
def predict(row, coefficients):
yhat = coefficients[0]
for i in range(len(row)-1):
yhat += coefficients[i + 1] * row[i]
return 1.0 / (1.0 + exp(-yhat))
我們可以設計一個小數據集來測試我們的 predict() 函數。
X1 X2 Y
2.7810836 2.550537003 0
1.465489372 2.362125076 0
3.396561688 4.400293529 0
1.38807019 1.850220317 0
3.06407232 3.005305973 0
7.627531214 2.759262235 1
5.332441248 2.088626775 1
6.922596716 1.77106367 1
8.675418651 -0.242068655 1
7.673756466 3.508563011 1
下面是數據集的散點圖,不同顏色代表不同類別。
我們還可以使用先前準備的係數對該數據集進行預測。
把這些放在一起,我們可以測試下面的 predict() 函數。
# Make a prediction
from math import exp
# Make a prediction with coefficients
def predict(row, coefficients):
yhat = coefficients[0]
for i in range(len(row)-1):
yhat += coefficients[i + 1] * row[i]
return 1.0 / (1.0 + exp(-yhat))
# test predictions
dataset = [[2.7810836,2.550537003,0],
[1.465489372,2.362125076,0],
[3.396561688,4.400293529,0],
[1.38807019,1.850220317,0],
[3.06407232,3.005305973,0],
[7.627531214,2.759262235,1],
[5.332441248,2.088626775,1],
[6.922596716,1.77106367,1],
[8.675418651,-0.242068655,1],
[7.673756466,3.508563011,1]]
coef = [-0.406605464, 0.852573316, -1.104746259]
for row in dataset:
yhat = predict(row, coef)
print("Expected=%.3f, Predicted=%.3f [%d]" % (row[-1], yhat, round(yhat)))
有兩個輸入值(X1 和 X2)和三個係數(b0,b1 和 b2)。該模型的預測方程是:
y = 1.0 / (1.0 + e^(-(b0 + b1 * X1 + b2 * X2)))
或者,代入我們主觀選擇的具體系數值,方程為:
y = 1.0 / (1.0 + e^(-(-0.406605464 + 0.852573316 * X1 + -1.104746259 * X2)))
運行此函數,我們得到的預測值相當接近預期的輸出值(y),並且四捨五入時能預測出正確的類別。
Expected=0.000, Predicted=0.299 [0]
Expected=0.000, Predicted=0.146 [0]
Expected=0.000, Predicted=0.085 [0]
Expected=0.000, Predicted=0.220 [0]
Expected=0.000, Predicted=0.247 [0]
Expected=1.000, Predicted=0.955 [1]
Expected=1.000, Predicted=0.862 [1]
Expected=1.000, Predicted=0.972 [1]
Expected=1.000, Predicted=0.999 [1]
Expected=1.000, Predicted=0.905 [1]
現在我們已經準備好實現隨機梯度下降算法來優化係數值了。
2. 估計係數
我們可以使用隨機梯度下降來估計訓練集的係數值。
隨機梯度下降需要兩個參數:
學習速率(Learning Rate):用於限制每次迭代時每個係數的校正量。
迭代次數(Epochs):更新係數前遍歷訓練集數據的次數。
函數中有 3 層循環:
1. 每次迭代(epoch)的循環。
2. 每次迭代的訓練集數據的每一行的循環。
3. 每次迭代的每一行數據的每個係數的每次更新的循環。
就這樣,在每一次迭代中,我們更新訓練集中每一行數據的每個係數。係數的更新基於模型的訓練誤差值。這個誤差通過期望輸出值(真實的因變量)與估計係數確定的預測值之間的差來計算。
每一個輸入屬性(自變量)對應一個係數,這些係數在迭代中不斷更新,例如:
b1(t+1) = b1(t) + learning_rate * (y(t) - yhat(t)) * yhat(t) * (1 - yhat(t)) * x1(t)
列表開頭的特殊係數(也稱為截距)以類似方式更新,除了與輸入值無關:
b0(t+1) = b0(t) + learning_rate * (y(t) - yhat(t)) * yhat(t) * (1 - yhat(t))
現在我們可以把這所有一切放在一起。下面是一個名為 coefficients_sgd() 的函數,它使用隨機梯度下降計算訓練集的係數值。
# Estimate logistic regression coefficients using stochastic gradient descent
def coefficients_sgd(train, l_rate, n_epoch):
coef = [0.0 for i in range(len(train[0]))]
for epoch in range(n_epoch):
sum_error = 0
for row in train:
yhat = predict(row, coef)
error = row[-1] - yhat
sum_error += error**2
coef[0] = coef[0] + l_rate * error * yhat * (1.0 - yhat)
for i in range(len(row)-1):
coef[i + 1] = coef[i + 1] + l_rate * error * yhat * (1.0 - yhat) * row[i]
print('>epoch=%d, lrate=%.3f, error=%.3f' % (epoch, l_rate, sum_error))
return coef
此外,每次迭代我們記錄誤差平方和 SSE(一個正值),以便我們在每個外循環開始時可以 print 出結果。
我們可以用上面的小數據集測試這個函數。
from math import exp
# Make a prediction with coefficients
def predict(row, coefficients):
yhat = coefficients[0]
for i in range(len(row)-1):
yhat += coefficients[i + 1] * row[i]
return 1.0 / (1.0 + exp(-yhat))
# Estimate logistic regression coefficients using stochastic gradient descent
def coefficients_sgd(train, l_rate, n_epoch):
coef = [0.0 for i in range(len(train[0]))]
for epoch in range(n_epoch):
sum_error = 0
for row in train:
yhat = predict(row, coef)
error = row[-1] - yhat
sum_error += error**2
coef[0] = coef[0] + l_rate * error * yhat * (1.0 - yhat)
for i in range(len(row)-1):
coef[i + 1] = coef[i + 1] + l_rate * error * yhat * (1.0 - yhat) * row[i]
print('>epoch=%d, lrate=%.3f, error=%.3f' % (epoch, l_rate, sum_error))
return coef
# Calculate coefficients
dataset = [[2.7810836,2.550537003,0],
[1.465489372,2.362125076,0],
[3.396561688,4.400293529,0],
[1.38807019,1.850220317,0],
[3.06407232,3.005305973,0],
[7.627531214,2.759262235,1],
[5.332441248,2.088626775,1],
[6.922596716,1.77106367,1],
[8.675418651,-0.242068655,1],
[7.673756466,3.508563011,1]]
l_rate = 0.3
n_epoch = 100
coef = coefficients_sgd(dataset, l_rate, n_epoch)
print(coef)
我們使用更大的學習速率 0.3,並循環 100 次迭代來訓練模型,或將係數更新 100 次。
運行該示例代碼,每次迭代時會 print 出該次迭代的誤差平方和以及該迭代確定的最優係數。
>epoch=95, lrate=0.300, error=0.023
>epoch=96, lrate=0.300, error=0.023
>epoch=97, lrate=0.300, error=0.023
>epoch=98, lrate=0.300, error=0.023
>epoch=99, lrate=0.300, error=0.022
[-0.8596443546618897, 1.5223825112460005, -2.218700210565016]
你可以看到誤差持續下降,甚至在最後一次迭代。我們可以訓練更長的時間(更多次迭代)或增加每次迭代更新係數的程度(更高的學習率)。
測試這些代碼,看看你有什麼新想法。
現在,讓我們將此算法應用於實際數據集。
3. 糖尿病數據集預測
在本節中,我們將使用隨機梯度下降算法對糖尿病數據集進行 logistic 回歸模型訓練。
該示例假定數據集的 CSV 副本位於當前工作目錄中,文件名為 pima-indians-diabetes.csv。
首先加載數據集,將字符串值轉換為數字,並將每個列標準化為 0 到 1 範圍內的值。這是通過輔助函數 load_csv()和 str_column_to_float()來加載和準備數據集以及 dataset_minmax()和 normalize_dataset()來標準化的。
我們將使用 k 折交叉驗證(k-fold cross validation)來估計學習到的模型在未知數據上的預測效果。這意味著我們將構建和評估 k 個模型,並將預測效果的平均值作為模型的評價標準。分類準確率將用於評估每個模型。這些過程由輔助函數 cross_validation_split(),accuracy_metric() 和 evaluate_algorithm() 提供。
我們將使用上面創建的 predict()、coefficients_sgd() 函數和一個新的 logistic_regression() 函數來訓練模型。
下面是完整示例:
# Logistic Regression on Diabetes Dataset
from random import seed
from random import randrange
from csv import reader
from math import exp
# Load a CSV file
def load_csv(filename):
dataset = list()
with open(filename, 'r') as file:
csv_reader = reader(file)
for row in csv_reader:
if not row:
continue
dataset.append(row)
return dataset
# Convert string column to float
def str_column_to_float(dataset, column):
for row in dataset:
row[column] = float(row[column].strip())
# Find the min and max values for each column
def dataset_minmax(dataset):
minmax = list()
for i in range(len(dataset[0])):
col_values = [row[i] for row in dataset]
value_min = min(col_values)
value_max = max(col_values)
minmax.append([value_min, value_max])
return minmax
# Rescale dataset columns to the range 0-1
def normalize_dataset(dataset, minmax):
for row in dataset:
for i in range(len(row)):
row[i] = (row[i] - minmax[i][0]) / (minmax[i][1] - minmax[i][0])
# Split a dataset into k folds
def cross_validation_split(dataset, n_folds):
dataset_split = list()
dataset_copy = list(dataset)
fold_size = len(dataset) / n_folds
for i in range(n_folds):
fold = list()
while len(fold) < fold_size:
index = randrange(len(dataset_copy))
fold.append(dataset_copy.pop(index))
dataset_split.append(fold)
return dataset_split
# Calculate accuracy percentage
def accuracy_metric(actual, predicted):
correct = 0
for i in range(len(actual)):
if actual[i] == predicted[i]:
correct += 1
return correct / float(len(actual)) * 100.0
# Evaluate an algorithm using a cross validation split
def evaluate_algorithm(dataset, algorithm, n_folds, *args):
folds = cross_validation_split(dataset, n_folds)
scores = list()
for fold in folds:
train_set = list(folds)
train_set.remove(fold)
train_set = sum(train_set, [])
test_set = list()
for row in fold:
row_copy = list(row)
test_set.append(row_copy)
row_copy[-1] = None
predicted = algorithm(train_set, test_set, *args)
actual = [row[-1] for row in fold]
accuracy = accuracy_metric(actual, predicted)
scores.append(accuracy)
return scores
# Make a prediction with coefficients
def predict(row, coefficients):
yhat = coefficients[0]
for i in range(len(row)-1):
yhat += coefficients[i + 1] * row[i]
return 1.0 / (1.0 + exp(-yhat))
# Estimate logistic regression coefficients using stochastic gradient descent
def coefficients_sgd(train, l_rate, n_epoch):
coef = [0.0 for i in range(len(train[0]))]
for epoch in range(n_epoch):
for row in train:
yhat = predict(row, coef)
error = row[-1] - yhat
coef[0] = coef[0] + l_rate * error * yhat * (1.0 - yhat)
for i in range(len(row)-1):
coef[i + 1] = coef[i + 1] + l_rate * error * yhat * (1.0 - yhat) * row[i]
return coef
# Linear Regression Algorithm With Stochastic Gradient Descent
def logistic_regression(train, test, l_rate, n_epoch):
predictions = list()
coef = coefficients_sgd(train, l_rate, n_epoch)
for row in test:
yhat = predict(row, coef)
yhat = round(yhat)
predictions.append(yhat)
return(predictions)
# Test the logistic regression algorithm on the diabetes dataset
seed(1)
# load and prepare data
filename = 'pima-indians-diabetes.csv'
dataset = load_csv(filename)
for i in range(len(dataset[0])):
str_column_to_float(dataset, i)
# normalize
minmax = dataset_minmax(dataset)
normalize_dataset(dataset, minmax)
# evaluate algorithm
n_folds = 5
l_rate = 0.1
n_epoch = 100
scores = evaluate_algorithm(dataset, logistic_regression, n_folds, l_rate, n_epoch)
print('Scores: %s' % scores)
print('Mean Accuracy: %.3f%%' % (sum(scores)/float(len(scores))))
令 k 折交叉驗證的 k 值為 5,每次迭代時用於評估的數量為 768/5 = 153.6 或剛好超過 150 個記錄。通過實驗選擇學習速率 0.1 和訓練迭代次數 100。
你可以嘗試其它的設置,看看模型的評估分數是否比我的更好。
運行此示例代碼,print 5 折交叉驗證的每一次的分數,最後 print 分類準確率的平均值。
可以看到,此算法的準確率大約為 77%,而如果我們使用零規則算法預測多數類,基線值為 65%,本算法的準確率高於基線值。
Scores: [73.20261437908496, 75.81699346405229, 75.81699346405229, 83.66013071895425, 78.43137254901961]
Mean Accuracy: 77.386%
擴展
以下是本教程的一些擴展,你可以自己來實現這些算法。
調整(Tune)示例中的參數。調整學習速率、迭代次數,甚至調整數據預處理方法,以改進數據集的準確率得分。
批處理(Batch)隨機梯度下降。改變隨機梯度下降算法,使得模型在歷次迭代中的更新能不斷積累,並且只在迭代結束後的一個批處理中更新係數。
其它分類問題。嘗試用該技術解決其它 UCI 機器學習庫中的二值分類問題。
回顧
在本教程中,你了解了如何使用隨機梯度下降算法實現 logistic 回歸。
你現在知道:
如何對多變量分類問題進行預測。
如何使用隨機梯度下降優化一組係數。
如何將該技術應用到真正的分類預測建模問題。
原文連結:https://machinelearningmastery.com/implement-logistic-regression-stochastic-gradient-descent-scratch-python/