add set leaf, constructor of tstats now rely on param

2014-08-24 16:07:59 -07:00 · 2014-08-24 16:07:59 -07:00 · 49e6575c86
commit 49e6575c86
parent ba9fbd380c
1 changed files with 2 additions and 2 deletions
--- a/demo/kaggle-higgs/speedtest.py
+++ b/demo/kaggle-higgs/speedtest.py
@ -50,8 +50,8 @@ num_round = 10
 print ('loading data end, start to boost trees')
 print ("training GBM from sklearn")
 tmp = time.time()
-gbm = GradientBoostingClassifier(n_estimators=num_round, max_depth=6, verbose=2)
-gbm.fit(data, label)
+#gbm = GradientBoostingClassifier(n_estimators=num_round, max_depth=6, verbose=2)
+#gbm.fit(data, label)
 print ("sklearn.GBM costs: %s seconds" % str(time.time() - tmp))
 #raw_input()
 print ("training xgboost")