ZBLOG

from sklearn preprocessing import PolynomialFeatures from sklearnpipeline import Pipeline from sklearnlinear model import LinearRegression from sklearn preprocessing import standardScaler def fit and predict(X train,ytrainXtest): #xtrain:numpy二维...

from sklearn.preprocessing import PolynomialFeatures from sklearn.pipeline import Pipeline from sklearn.linear_model import Ridge from sklearn.preprocessing import StandardScaler

def fit_and_predict(X_train, y_train, X_test):

#xtrain:numpy二维数组,由n个样本、k个特征组成的数据矩阵,形状是(n,k)。
#ytrain:numpy一维数组,由n个数据组成的标签,形状是(n,)。
#xtest:numpy二维数组,由m个测试、k个特征组成的数据矩阵,形状是(m,k)。

model = Pipeline([
    ('poly', PolynomialFeatures(6)),
    ('scaler', StandardScaler()),
    ('ridge', Ridge(alpha=0.5))
])

model.fit(X_train, y_train)

return model.predict(X_test)

#使用更大的alpha值可以减少模型过拟合的情况。同时也需要根据具体数据情况进行参数调整。

本站部分文章来源于网络,版权归原作者所有,如有侵权请联系站长删除。
转载请注明出处:https://golang.0voice.com/?id=4264

分享:
扫描分享到社交APP
上一篇
下一篇
发表列表
游客 游客
此处应有掌声~
评论列表

还没有评论,快来说点什么吧~

联系我们

在线咨询: 点击这里给我发消息

微信号:3007537140

上班时间: 10:30-22:30

关注我们
x

注册

已经有帐号?