首页 文章

GridsearchCV用于多项式回归

提问于
浏览
0

我是机器学习的新手并坚持这一点 .

当我试图在线性模型中实现多项式回归时,比如使用多个多项式范围(1,10)并得到不同的MSE . 我实际上使用 GridsearchCV 方法来找到多项式的最佳参数 .

from sklearn.model_selection import GridSearchCV

poly_grid = GridSearchCV(PolynomialRegression(), param_grid, cv=10, scoring='neg_mean_squared_error')

我不知道如何获得上述 PolynomialRegression() 估算器 . 我搜索的一个解决方案是:

import numpy as np
from sklearn.model_selection import GridSearchCV
from sklearn.pipeline import make_pipeline

def PolynomialRegression(degree=2, **kwargs):
    return make_pipeline(PolynomialFeatures(degree), LinearRegression(**kwargs))

param_grid = {'polynomialfeatures__degree': np.arange(10), 'linearregression__fit_intercept': [True, False], 'linearregression__normalize': [True, False]}

poly_grid = GridSearchCV(PolynomialRegression(), param_grid, cv=10, scoring='neg_mean_squared_error')

但它甚至没有产生任何结果 .

1 回答

相关问题