EL之GB(GBM)：利用GB对回归(性别属性编码+调2参)问题(整数值年龄预测)建模

EL之GB(GBM)：利用GB对回归(性别属性编码+调2参)问题(整数值年龄预测)建模

目录

输出结果

设计思路

核心代码

输出结果

T1、

T2、

设计思路

核心代码

  

   

    

     

    
    

     

      #T1
     
    
   

    

     

    
    

     

      nEst = 2000
     
    
   

    

     

    
    

     

      depth = 5
     
    
   

    

     

    
    

     

      learnRate = 0.003
     
    
   

    

     

    
    

     

      maxFeatures = None
     
    
   

    

     

    
    

     

      subsamp = 0.5
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      #T2
     
    
   

    

     

    
    

     

      # nEst = 2000
     
    
   

    

     

    
    

     

      # depth = 5
     
    
   

    

     

    
    

     

      # learnRate = 0.005
     
    
   

    

     

    
    

     

      # maxFeatures = 3
     
    
   

    

     

    
    

     

      # subsamp = 0.5
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      abaloneGBMModel = ensemble.GradientBoostingRegressor(n_estimators=nEst, max_depth=depth, 
     
    
   

    

     

    
    

     

       learning_rate=learnRate, max_features=maxFeatures,
     
    
   

    

     

    
    

     

       subsample=subsamp, loss='ls')
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      abaloneGBMModel.fit(xTrain, yTrain)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      # compute mse on test set
     
    
   

    

     

    
    

     

      msError = []
     
    
   

    

     

    
    

     

      predictions = abaloneGBMModel._staged_decision_function(xTest)
     
    
   

    

     

    
    

     

      for p in predictions:
     
    
   

    

     

    
    

     

       msError.append(mean_squared_error(yTest, p))
     
    
  
 

文章来源: yunyaniu.blog.csdn.net，作者：一个处女座的程序猿，版权归原作者所有，如需转载，请联系作者。

原文链接：yunyaniu.blog.csdn.net/article/details/86377011