From 49e6575c869bfc2ced93874071ef216bc813b14c Mon Sep 17 00:00:00 2001 From: tqchen Date: Sun, 24 Aug 2014 16:07:59 -0700 Subject: [PATCH] add set leaf, constructor of tstats now rely on param --- demo/kaggle-higgs/speedtest.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/demo/kaggle-higgs/speedtest.py b/demo/kaggle-higgs/speedtest.py index c5cc2fd29..92d745302 100755 --- 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")