Scikit-Learn แน่นอนไม่สนับสนุนการถดถอยแบบขั้นตอน นั่นเป็นเพราะสิ่งที่เป็นที่รู้จักกันทั่วไปว่า 'การถดถอยแบบขั้นตอน' เป็นอัลกอริทึมที่ขึ้นอยู่กับค่า p ของสัมประสิทธิ์การถดถอยเชิงเส้นและ scikit เรียนรู้จงใจหลีกเลี่ยงวิธีการอนุมานเพื่อการเรียนรู้แบบจำลอง ยิ่งไปกว่านั้น OLS ที่บริสุทธิ์เป็นเพียงหนึ่งในอัลกอริธึมการถดถอยจำนวนมากและจากมุมมองของการเรียนรู้ scikit มันไม่สำคัญมากหรือเป็นหนึ่งในสิ่งที่ดีที่สุด
อย่างไรก็ตามมีคำแนะนำบางอย่างสำหรับผู้ที่ยังต้องการวิธีที่ดีในการเลือกคุณสมบัติด้วยโมเดลเชิงเส้น:
- ใช้โมเดลที่เบาบางโดยเนื้อแท้เหมือนหรือ
ElasticNetLasso
- ทำให้ปกติคุณสมบัติของคุณเป็นปกติ
StandardScalerแล้วสั่งซื้อคุณสมบัติของคุณเพียงแค่model.coef_นี้ สำหรับตัวแปรอิสระที่สมบูรณ์แบบมันเทียบเท่ากับการเรียงลำดับตามค่า p ชั้นเรียนsklearn.feature_selection.RFEจะทำเพื่อคุณและRFECVจะประเมินจำนวนคุณลักษณะที่เหมาะสม
- R2
statsmodels
- ทำการเลือกอย่างไร้เดียงสาไปข้างหน้าหรือข้างหลังเพื่อเพิ่มการวัดที่คุณชื่นชอบให้มากที่สุดในการตรวจสอบความถูกต้องไขว้กัน
mlxtendแพคเกจที่เข้ากันได้กับ Scikit-Learn รองรับวิธีนี้สำหรับตัวประมาณและตัวชี้วัดใด ๆ
- หากคุณยังคงต้องการการถดถอยแบบขั้นตอนของวานิลลามันจะง่ายกว่า
statsmodelsเพราะจะคำนวณตามค่า p สำหรับคุณ การเลือกไปข้างหน้าย้อนกลับขั้นพื้นฐานอาจมีลักษณะเช่นนี้:
`` `
from sklearn.datasets import load_boston
import pandas as pd
import numpy as np
import statsmodels.api as sm
data = load_boston()
X = pd.DataFrame(data.data, columns=data.feature_names)
y = data.target
def stepwise_selection(X, y,
initial_list=[],
threshold_in=0.01,
threshold_out = 0.05,
verbose=True):
""" Perform a forward-backward feature selection
based on p-value from statsmodels.api.OLS
Arguments:
X - pandas.DataFrame with candidate features
y - list-like with the target
initial_list - list of features to start with (column names of X)
threshold_in - include a feature if its p-value < threshold_in
threshold_out - exclude a feature if its p-value > threshold_out
verbose - whether to print the sequence of inclusions and exclusions
Returns: list of selected features
Always set threshold_in < threshold_out to avoid infinite looping.
See https://en.wikipedia.org/wiki/Stepwise_regression for the details
"""
included = list(initial_list)
while True:
changed=False
# forward step
excluded = list(set(X.columns)-set(included))
new_pval = pd.Series(index=excluded)
for new_column in excluded:
model = sm.OLS(y, sm.add_constant(pd.DataFrame(X[included+[new_column]]))).fit()
new_pval[new_column] = model.pvalues[new_column]
best_pval = new_pval.min()
if best_pval < threshold_in:
best_feature = new_pval.argmin()
included.append(best_feature)
changed=True
if verbose:
print('Add {:30} with p-value {:.6}'.format(best_feature, best_pval))
# backward step
model = sm.OLS(y, sm.add_constant(pd.DataFrame(X[included]))).fit()
# use all coefs except intercept
pvalues = model.pvalues.iloc[1:]
worst_pval = pvalues.max() # null if pvalues is empty
if worst_pval > threshold_out:
changed=True
worst_feature = pvalues.argmax()
included.remove(worst_feature)
if verbose:
print('Drop {:30} with p-value {:.6}'.format(worst_feature, worst_pval))
if not changed:
break
return included
result = stepwise_selection(X, y)
print('resulting features:')
print(result)
ตัวอย่างนี้จะพิมพ์ผลลัพธ์ต่อไปนี้:
Add LSTAT with p-value 5.0811e-88
Add RM with p-value 3.47226e-27
Add PTRATIO with p-value 1.64466e-14
Add DIS with p-value 1.66847e-05
Add NOX with p-value 5.48815e-08
Add CHAS with p-value 0.000265473
Add B with p-value 0.000771946
Add ZN with p-value 0.00465162
resulting features:
['LSTAT', 'RM', 'PTRATIO', 'DIS', 'NOX', 'CHAS', 'B', 'ZN']