xgboost

Type Members

1. case class FVecTensorImpl(tensor: Tensor[Double]) extends FVec with Product with Serializable

   

   Created by hollinwilkins on 9/16/17.

2. case class XGBoostBinaryClassificationModel(predictor: Predictor, booster: Option[Array[Byte]], numFeatures: Int, outputMargin: Boolean) extends XGBoostClassificationModelBase with Product with Serializable

   

3. case class XGBoostClassification(uid: String = ..., shape: NodeShape, model: XGBoostClassificationModel) extends MultiTransformer with Product with Serializable

   

   Created by hollinwilkins on 9/16/17.

4. case class XGBoostClassificationModel(impl: XGBoostClassificationModelBase) extends ProbabilisticClassificationModel with Product with Serializable

   

5. trait XGBoostClassificationModelBase extends ProbabilisticClassificationModel

   

6. case class XGBoostMultinomialClassificationModel(predictor: Predictor, booster: Option[Array[Byte]], numClasses: Int, numFeatures: Int, outputMargin: Boolean) extends XGBoostClassificationModelBase with Product with Serializable

   

7. case class XGBoostRegression(uid: String = ..., shape: NodeShape, model: XGBoostRegressionModel) extends SimpleTransformer with Product with Serializable

   

   Created by hollinwilkins on 9/16/17.

8. case class XGBoostRegressionModel(predictor: Predictor, booster: Option[Array[Byte]] = None, numFeatures: Int) extends Model with Product with Serializable

   

   Created by hollinwilkins on 9/16/17.