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

add set leaf, constructor of tstats now rely on param

Browse Source
This commit is contained in:
tqchen 2014-08-24 16:07:59 -07:00
parent ba9fbd380c
commit 49e6575c86
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 ('loading data end, start to boost trees')
print ("training GBM from sklearn") print ("training GBM from sklearn")
tmp = time.time() tmp = time.time()
gbm = GradientBoostingClassifier(n_estimators=num_round, max_depth=6, verbose=2) #gbm = GradientBoostingClassifier(n_estimators=num_round, max_depth=6, verbose=2)
gbm.fit(data, label) #gbm.fit(data, label)
print ("sklearn.GBM costs: %s seconds" % str(time.time() - tmp)) print ("sklearn.GBM costs: %s seconds" % str(time.time() - tmp))
#raw_input() #raw_input()
print ("training xgboost") print ("training xgboost")