ml.dmlc.xgboost4j.java

Class RabitTracker

java.lang.Object

ml.dmlc.xgboost4j.java.RabitTracker

All Implemented Interfaces:

Thread.UncaughtExceptionHandler, ITracker

public class RabitTracker
extends Object
implements ITracker

Java implementation of the Rabit tracker to coordinate distributed workers.

Constructor Summary

Constructors

Constructor and Description
RabitTracker(int numWorkers)
RabitTracker(int numWorkers, String hostIp)
RabitTracker(int numWorkers, String hostIp, int port, int timeout)

Method Summary

Modifier and Type	Method and Description
Map<String,Object>	getWorkerArgs() Get environments that can be used to pass to worker.
boolean	start()
void	stop()
void	uncaughtException(Thread t, Throwable e)
void	waitFor(long timeout)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

RabitTracker

public RabitTracker(int numWorkers)
             throws XGBoostError

Throws:

XGBoostError

RabitTracker

public RabitTracker(int numWorkers,
                    String hostIp)
             throws XGBoostError

Throws:

XGBoostError

RabitTracker

public RabitTracker(int numWorkers,
                    String hostIp,
                    int port,
                    int timeout)
             throws XGBoostError

Throws:

XGBoostError

Method Detail

uncaughtException

public void uncaughtException(Thread t,
                              Throwable e)

Specified by:

uncaughtException in interface Thread.UncaughtExceptionHandler

getWorkerArgs

public Map<String,Object> getWorkerArgs()
                                 throws XGBoostError

Get environments that can be used to pass to worker.

Specified by:

getWorkerArgs in interface ITracker

Returns:

The environment settings.

Throws:

XGBoostError

stop

public void stop()
          throws XGBoostError

Specified by:

stop in interface ITracker

Throws:

XGBoostError

start

public boolean start()
              throws XGBoostError

Specified by:

start in interface ITracker

Throws:

XGBoostError

waitFor

public void waitFor(long timeout)
             throws XGBoostError

Specified by:

waitFor in interface ITracker

Throws:

XGBoostError