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

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

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

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