Fix MPI build. (#6403)

This commit is contained in:
Jiaming Yuan
2020-11-21 13:38:21 +08:00
committed by GitHub
parent 2ce2a1a4d8
commit 42d31d9dcb
4 changed files with 52 additions and 32 deletions

View File

@@ -0,0 +1,13 @@
#!/bin/bash
rm -f *.model*
export DMLC_SUBMIT_CLUSTER=mpi
submit="timeout 5 python ../../dmlc-core/tracker/dmlc-submit"
echo "====== 1. Basic distributed test with Python ======"
$submit --cluster=local --num-workers=3 python test_basic.py
echo "====== 2. Regression test for issue #3402 ======"
$submit --cluster=local --num-workers=2 --worker-cores=1 python test_issue3402.py

View File

@@ -65,7 +65,7 @@ y = [1, 0]
dtrain = xgb.DMatrix(X, label=y)
param = {'max_depth': 2, 'eta': 1, 'silent': 1, 'objective': 'binary:logistic' }
param = {'max_depth': 2, 'eta': 1, 'objective': 'binary:logistic' }
watchlist = [(dtrain,'train')]
num_round = 2
bst = xgb.train(param, dtrain, num_round, watchlist)