Package

ml.dmlc.xgboost4j.scala

spark

Permalink

package spark

Visibility
  1. Public
  2. All

Type Members

  1. case class TrackerConf(workerConnectionTimeout: Long, trackerImpl: String) extends Product with Serializable

    Permalink

    Rabit tracker configurations.

    Rabit tracker configurations.

    workerConnectionTimeout

    The timeout for all workers to connect to the tracker. Set timeout length to zero to disable timeout. Use a finite, non-zero timeout value to prevent tracker from hanging indefinitely (in milliseconds) (supported by "scala" implementation only.)

    trackerImpl

    Choice between "python" or "scala". The former utilizes the Java wrapper of the Python Rabit tracker (in dmlc_core), whereas the latter is implemented in Scala without Python components, and with full support of timeouts. The Scala implementation is currently experimental, use at your own risk.

  2. class XGBoostClassificationModel extends XGBoostModel

    Permalink

    class of the XGBoost model used for classification task

  3. class XGBoostEstimator extends Predictor[Vector, XGBoostEstimator, XGBoostModel] with LearningTaskParams with GeneralParams with BoosterParams with MLWritable

    Permalink

    XGBoost Estimator to produce a XGBoost model

  4. abstract class XGBoostModel extends PredictionModel[Vector, XGBoostModel] with BoosterParams with Serializable with Params with MLWritable

    Permalink

    the base class of XGBoostClassificationModel and XGBoostRegressionModel

  5. class XGBoostRegressionModel extends XGBoostModel

    Permalink

    class of XGBoost model used for regression task

  6. class XGBoostTrainingSummary extends Serializable

    Permalink

Value Members

  1. object CheckpointManager

    Permalink
  2. object DataUtils extends Serializable

    Permalink
  3. object TrackerConf extends Serializable

    Permalink
  4. object XGBoost extends Serializable

    Permalink
  5. object XGBoostEstimator extends MLReadable[XGBoostEstimator] with Serializable

    Permalink
  6. object XGBoostModel extends MLReadable[XGBoostModel] with Serializable

    Permalink
  7. package params

    Permalink

Ungrouped