Package biz.k11i.xgboost.spark
Class SparkModelParam
- java.lang.Object
-
- biz.k11i.xgboost.spark.SparkModelParam
-
- All Implemented Interfaces:
java.io.Serializable
public class SparkModelParam extends java.lang.Object implements java.io.Serializable
- See Also:
- Serialized Form
-
-
Field Summary
Fields Modifier and Type Field Description (package private) java.lang.String
featureCol
(package private) java.lang.String
labelCol
static java.lang.String
MODEL_TYPE_CLS
static java.lang.String
MODEL_TYPE_REG
(package private) java.lang.String
modelType
(package private) java.lang.String
predictionCol
(package private) java.lang.String
rawPredictionCol
(package private) double[]
thresholds
-
Constructor Summary
Constructors Constructor Description SparkModelParam(java.lang.String modelType, java.lang.String featureCol, ModelReader reader)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description java.lang.String
getFeatureCol()
java.lang.String
getLabelCol()
java.lang.String
getModelType()
java.lang.String
getPredictionCol()
java.lang.String
getRawPredictionCol()
double[]
getThresholds()
-
-
-
Field Detail
-
MODEL_TYPE_CLS
public static final java.lang.String MODEL_TYPE_CLS
- See Also:
- Constant Field Values
-
MODEL_TYPE_REG
public static final java.lang.String MODEL_TYPE_REG
- See Also:
- Constant Field Values
-
modelType
final java.lang.String modelType
-
featureCol
final java.lang.String featureCol
-
labelCol
final java.lang.String labelCol
-
predictionCol
final java.lang.String predictionCol
-
rawPredictionCol
final java.lang.String rawPredictionCol
-
thresholds
final double[] thresholds
-
-
Constructor Detail
-
SparkModelParam
public SparkModelParam(java.lang.String modelType, java.lang.String featureCol, ModelReader reader) throws java.io.IOException
- Throws:
java.io.IOException
-
-
Method Detail
-
getModelType
public java.lang.String getModelType()
-
getFeatureCol
public java.lang.String getFeatureCol()
-
getLabelCol
public java.lang.String getLabelCol()
-
getPredictionCol
public java.lang.String getPredictionCol()
-
getRawPredictionCol
public java.lang.String getRawPredictionCol()
-
getThresholds
public double[] getThresholds()
-
-