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:num...

修改后的代码:

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

def fit_and_predict(X_train, y_train, X_test):

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

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

model.fit(X_train, y_train)

return model.predict(X_test)

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

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

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

联系我们

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

微信号:3007537140

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

关注我们
x

注册

已经有帐号?