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