Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

chg higgs back

Browse Source
This commit is contained in:
tqchen 2014-08-24 16:09:13 -07:00
parent 4889b40abc
commit 19447cdb12
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
demo/kaggle-higgs/speedtest.py
View File

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