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 final String	MODEL_TYPE_CLS
static final String	MODEL_TYPE_REG

Constructor Summary

Constructors

Constructor	Description
SparkModelParam(String modelType, String featureCol, ModelReader reader)

Method Summary

Modifier and Type	Method	Description
String	getFeatureCol()
String	getLabelCol()
String	getModelType()
String	getPredictionCol()
String	getRawPredictionCol()
double[]	getThresholds()

Methods inherited from class Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Details

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 Details

SparkModelParam

public SparkModelParam(String modelType,
 String featureCol,
 ModelReader reader)
                throws IOException

Throws:

IOException

Method Details

getModelType

public String getModelType()

getFeatureCol

public String getFeatureCol()

getLabelCol

public String getLabelCol()

getPredictionCol

public String getPredictionCol()

getRawPredictionCol

public String getRawPredictionCol()

getThresholds

public double[] getThresholds()