Update rabit

This commit is contained in:
tqchen
2016-03-01 21:02:35 -08:00
parent 0515e4ec28
commit ced6d45e01
5 changed files with 22 additions and 6 deletions

View File

@@ -4,6 +4,9 @@ import scipy.sparse
import pickle
import xgboost as xgb
# always call this before using distributed module
xgb.rabit.init()
# Load file, file will be automatically sharded in distributed mode.
dtrain = xgb.DMatrix('../../demo/data/agaricus.txt.train')
dtest = xgb.DMatrix('../../demo/data/agaricus.txt.test')