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")