[JVM-PKG] add distributed test simple case

2016-03-02 22:27:55 -08:00
parent 5c9e50148a
commit c428a93adc
3 changed files with 57 additions and 1 deletions
--- a/jvm-packages/xgboost4j/src/main/java/ml/dmlc/xgboost4j/XGBoost.java
+++ b/jvm-packages/xgboost4j/src/main/java/ml/dmlc/xgboost4j/XGBoost.java
@@ -96,7 +96,9 @@ public class XGBoost {
        } else {
          evalInfo = booster.evalSet(evalMats, evalNames, iter);
        }
-        logger.info(evalInfo);
+        if (Rabit.getRank() == 0) {
+          Rabit.trackerPrint(evalInfo + '\n');
+        }
      }
      booster.saveRabitCheckpoint();
      version += 1;