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 and Description
static java.lang.String	MODEL_TYPE_CLS
static java.lang.String	MODEL_TYPE_REG

Constructor Summary

Constructors

Constructor and Description
SparkModelParam(java.lang.String modelType, java.lang.String featureCol, ModelReader reader)

Method Summary

Methods

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

Methods inherited from class java.lang.Object

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

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

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()