From 19447cdb12686c6e70a46363f2b4f684e20a7cd9 Mon Sep 17 00:00:00 2001 From: tqchen Date: Sun, 24 Aug 2014 16:09:13 -0700 Subject: [PATCH] chg higgs back --- 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 92d745302..c5cc2fd29 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")