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

minor changes

Browse Source
This commit is contained in:
tqchen 2014-05-16 18:19:57 -07:00
parent ae70b9b152
commit c28a1be34c
3 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
demo/kaggle-higgs/README.md
View File

@ -7,4 +7,4 @@ run ./run.sh
you need to compile xgboost python model in ../../python you need to compile xgboost python model in ../../python
speedtest.py compares xgboost's speed on this dataset with sklearn

4
demo/kaggle-higgs/run.sh
View File

@ -1,4 +1,4 @@
#!/bin/bash #!/bin/bash
./higgs-numpy.py python higgs-numpy.py
./higgs-pred.py python higgs-pred.py

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

@ -1,4 +1,4 @@
#!/usr/local/bin/python #!/usr/bin/python
# this is the example script to use xgboost to train # this is the example script to use xgboost to train
import sys import sys
import numpy as np import numpy as np
@ -52,7 +52,7 @@ 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 "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"
threads = [1, 2, 4, 16] threads = [1, 2, 4, 16]