码迷,mamicode.com
首页 > 编程语言 > 详细

算法实现之python篇

时间:2015-07-05 23:50:23      阅读:166      评论:0      收藏:0      [点我收藏+]

标签:

Python source code: gradient_boosting_regression.py

from sklearn import ensemblefrom sklearn.metrics import mean_squared_error

# Fit regression model
params = {n_estimators: 500, max_depth: 4, min_samples_split: 1,
          learning_rate: 0.01, loss: ls}
clf = ensemble.GradientBoostingRegressor(**params)

clf.fit(X_train, y_train)
mse = mean_squared_error(y_test, clf.predict(X_test))
print("MSE: %.4f" % mse)

 

算法实现之python篇

标签:

原文地址:http://www.cnblogs.com/chaofn/p/4623152.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!