#!/bin/bash
if [[ $# -ne 1 ]]
then
    echo "Usage: nprocess"
    exit -1
fi

rm -rf train-machine.row* *.model
k=$1
# make machine data
cd ../../demo/regression/
python mapfeat.py
python mknfold.py machine.txt 1
cd -

# split the lib svm file into k subfiles
python splitrows.py ../../demo/regression/machine.txt.train train-machine $k

# run xgboost mpi
mpirun -n $k ../../xgboost-mpi machine-row.conf dsplit=row num_round=3

# run xgboost-mpi save model 0001, continue to run from existing model
mpirun -n $k ../../xgboost-mpi machine-row.conf dsplit=row num_round=1
mpirun -n $k ../../xgboost-mpi machine-row.conf dsplit=row num_round=2 model_in=0001.model