python sklearn庫實現多元線性回歸模型的建立,并進行評估和調優

1.數據集的導入以及自變量和因變量的選取

作為機器學習的入門知識點和模型,網上有很多sklearn庫實現線性回歸的例子。不過在這里我會從最開始的模型訓練到最后的模型調優,統一地進行一個練習,以便對線性回歸有個更深入淺出的理解。

首先我們先導入一些相關的包:

import pandas as pd
import matplotlib.pyplot as plt
import matplotlib as mpl
from sklearn.linear_model import LinearRegression
from sklearn.model_selection import cross_val_predict, train_test_split
from sklearn import datasets
%matplotlib inline
mpl.rcParams['font.family'] = ['sans-serif']
mpl.rcParams['font.sans-serif'] = ['SimHei']
mpl.rcParams['axes.unicode_minus']=False

作為回歸模型練習的數據集,就從經典的波士頓房價數據集來進行訓練就好,這也是sklearn庫中自帶的小數據集。在這里,作為自變量X的就是數據集中的feature_names維度,總共有13個維度,所以這是一個多元線性回歸模型;因變量y就是我們數據集中的target維度——房價。

data = datasets.load_boston()
df = pd.DataFrame(data.data, columns=data.feature_names)
target = pd.DataFrame(data.target, columns=['MEDV'])
X = df
y = target

2.劃分訓練集和測試集

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.5, random_state=1)
print(X_train.shape)
print(X_test.shape)
1.png

可以看到由于參數test_size是0.5,所以訓練集和測試集對半分,維度都是13.

3.模型的訓練

劃分好訓練集和測試集后,就是對模型進行訓練了,然后我們輸出截距和權重系數,看下模型

lr = LinearRegression()
lr.fit(X_train, y_train)

print(lr.coef_)
print(lr.intercept_)
2.png

因此,波士頓房價的多元線性模型就此誕生,下面就要評估下這兩個系數擬合的模型能否相對準確地預測出因變量y——房價了。

4.模型的評估

對于多元線性回歸模型的評估,主要有四種指標,這里我選用了RMSE——均方差根,數值越小說明模型擬合度越高

y_pred = lr.predict(X_test)
from sklearn import metrics
MSE = metrics.mean_squared_error(y_test, y_pred)
RMSE = np.sqrt(metrics.mean_squared_error(y_test, y_pred))

print('MSE:',MSE)
print('RMSE:',RMSE)
3.png

圖中可以看到RMSE有4.779,說明擬合度還不夠高,我們通過可視化來看一下訓練后的預測和真實值之間的差異:

plt.figure(figsize=(15,5))
plt.plot(range(len(y_test)), y_test, 'r', label='測試數據')
plt.plot(range(len(y_test)), y_pred, 'b', label='預測數據')
plt.legend()
預測數據和真實數據的折線圖.png

可以看到許多地方,紅線會明顯超出藍線,說明的確,我們的模型擬合度不夠,再換個散點圖來更直觀地看一下:

plt.scatter(y_test, y_pred)
plt.plot([y_test.min(),y_test.max()], [y_test.min(),y_test.max()], 'k--')
plt.xlabel('真實值')
plt.ylabel('預測值')
預測值和真實值scatter.png

圖中我們可以看到,如果完全擬合,散點應該和直線相重合,這里發現,y_test=50的地方,有較多的異常值,而線性回歸模型的一大缺點就是對異常值很敏感,會極大影響模型的準確性,因此,下一步,我們就根據這一點,對模型進行優化

5.模型調優

# 去除MEDV=50的異常值
drop_index = y[y['MEDV']==50].index.values
X = X.drop(drop_index)
y = y.drop(drop_index)
# 這里劃分訓練集和測試集的參數random_state都是1,表示隨機分配的數據是同一組
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.25, random_state=1)
# 對訓練集進行訓練
lr = LinearRegression()
lr.fit(X_train, y_train)
# 對測試集進行預測
y_pred = lr.predict(X_test)
RMSE = np.sqrt(metrics.mean_squared_error(y_test, y_pred))
print('RMSE:',RMSE)
4.png

可以看到,該模型的RMSE變成了3.573,大大提高了模型的準確率。這里我對回歸模型的調優主要是兩個方面:1. 增加了訓練集的數量,提高訓練準確度;2. 去除異常值,提高準確度。

plt.scatter(y_test, y_pred)
plt.plot([y_test.min(),y_test.max()], [y_test.min(),y_test.max()], 'k--')
plt.xlabel('真實值')
plt.ylabel('預測值')
調優后的回歸模型.png

總結:

線性回歸模型的評估可以根據RMSE來判斷;調優模型主要通過放大訓練集數量以及排除異常值來進行;LinearRegression本身的參數只有四個,大多數情況下也不需要進行參數調整,默認的就OK。

?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 230,527評論 6 544
  • 序言:濱河連續發生了三起死亡事件,死亡現場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發現死者居然都...
    沈念sama閱讀 99,687評論 3 429
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 178,640評論 0 383
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 63,957評論 1 318
  • 正文 為了忘掉前任,我火速辦了婚禮,結果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 72,682評論 6 413
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發上,一...
    開封第一講書人閱讀 56,011評論 1 329
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 44,009評論 3 449
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 43,183評論 0 290
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當地人在樹林里發現了一具尸體,經...
    沈念sama閱讀 49,714評論 1 336
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 41,435評論 3 359
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發現自己被綠了。 大學時的朋友給我發了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 43,665評論 1 374
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 39,148評論 5 365
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質發生泄漏。R本人自食惡果不足惜,卻給世界環境...
    茶點故事閱讀 44,838評論 3 350
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 35,251評論 0 28
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 36,588評論 1 295
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 52,379評論 3 400
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 48,627評論 2 380

推薦閱讀更多精彩內容