Package biz.k11i.xgboost.spark
Class SparkModelParam
java.lang.Object
biz.k11i.xgboost.spark.SparkModelParam
- All Implemented Interfaces:
Serializable
- See Also:
-
Field Summary
Fields -
Constructor Summary
Constructors -
Method Summary
Modifier and TypeMethodDescriptiondouble[]
-
Field Details
-
MODEL_TYPE_CLS
- See Also:
-
MODEL_TYPE_REG
- See Also:
-
modelType
-
featureCol
-
labelCol
-
predictionCol
-
rawPredictionCol
-
thresholds
final double[] thresholds
-
-
Constructor Details
-
SparkModelParam
- Throws:
IOException
-
-
Method Details
-
getModelType
-
getFeatureCol
-
getLabelCol
-
getPredictionCol
-
getRawPredictionCol
-
getThresholds
public double[] getThresholds()
-