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):

  1. #xtrain:numpy二维数组,由n个样本、k个特征组成的数据矩阵,形状是(n,k)。
  2. #ytrain:numpy一维数组,由n个数据组成的标签,形状是(n,)。
  3. #xtest:numpy二维数组,由m个测试、k个特征组成的数据矩阵,形状是(m,k)。
  4. model = Pipeline([
  5. ('poly', PolynomialFeatures(6)),
  6. ('scaler', StandardScaler()),
  7. ('ridge', Ridge(alpha=0.5))
  8. ])
  9. model.fit(X_train, y_train)
  10. return model.predict(X_test)

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

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

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

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

联系我们

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

微信号:3007537140

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

关注我们
x

注册

已经有帐号?