diff --git a/demo/kaggle-higgs/README.md b/demo/kaggle-higgs/README.md index b2ec971bd..2736b1326 100644 --- a/demo/kaggle-higgs/README.md +++ b/demo/kaggle-higgs/README.md @@ -7,4 +7,4 @@ run ./run.sh you need to compile xgboost python model in ../../python - +speedtest.py compares xgboost's speed on this dataset with sklearn diff --git a/demo/kaggle-higgs/run.sh b/demo/kaggle-higgs/run.sh index e6b5d91fa..c69426c25 100755 --- a/demo/kaggle-higgs/run.sh +++ b/demo/kaggle-higgs/run.sh @@ -1,4 +1,4 @@ #!/bin/bash -./higgs-numpy.py -./higgs-pred.py \ No newline at end of file +python higgs-numpy.py +python higgs-pred.py \ No newline at end of file diff --git a/demo/kaggle-higgs/speedtest.py b/demo/kaggle-higgs/speedtest.py index d17c2680b..212389c01 100755 --- a/demo/kaggle-higgs/speedtest.py +++ b/demo/kaggle-higgs/speedtest.py @@ -1,4 +1,4 @@ -#!/usr/local/bin/python +#!/usr/bin/python # this is the example script to use xgboost to train import sys import numpy as np @@ -52,7 +52,7 @@ print "training GBM from sklearn" tmp = time.time() gbm = GradientBoostingClassifier(n_estimators=num_round, max_depth=6, verbose=2) 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() print "training xgboost" threads = [1, 2, 4, 16]