Class

com.salesforce.op.stages.impl.classification

OpDecisionTreeClassifier

Related Doc: package classification

Permalink

class OpDecisionTreeClassifier extends OpPredictorWrapper[DecisionTreeClassifier, DecisionTreeClassificationModel] with OpDecisionTreeClassifierParams

Wrapper for spark decision tree classifier org.apache.spark.ml.classification.DecisionTreeClassifier

Linear Supertypes
OpDecisionTreeClassifierParams, DecisionTreeClassifierParams, TreeClassifierParams, DecisionTreeParams, HasSeed, HasCheckpointInterval, PredictorParams, HasPredictionCol, HasFeaturesCol, HasLabelCol, OpPredictorWrapper[DecisionTreeClassifier, DecisionTreeClassificationModel], SparkWrapperParams[DecisionTreeClassifier], OpPipelineStage2[RealNN, OPVector, Prediction], HasIn2, HasIn1, OpPipelineStage[Prediction], OpPipelineStageBase, MLWritable, OpPipelineStageParams, InputParams, Estimator[OpPredictorWrapperModel[DecisionTreeClassificationModel]], PipelineStage, Logging, Params, Serializable, Serializable, Identifiable, AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. OpDecisionTreeClassifier
  2. OpDecisionTreeClassifierParams
  3. DecisionTreeClassifierParams
  4. TreeClassifierParams
  5. DecisionTreeParams
  6. HasSeed
  7. HasCheckpointInterval
  8. PredictorParams
  9. HasPredictionCol
  10. HasFeaturesCol
  11. HasLabelCol
  12. OpPredictorWrapper
  13. SparkWrapperParams
  14. OpPipelineStage2
  15. HasIn2
  16. HasIn1
  17. OpPipelineStage
  18. OpPipelineStageBase
  19. MLWritable
  20. OpPipelineStageParams
  21. InputParams
  22. Estimator
  23. PipelineStage
  24. Logging
  25. Params
  26. Serializable
  27. Serializable
  28. Identifiable
  29. AnyRef
  30. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new OpDecisionTreeClassifier(uid: String = UID[OpDecisionTreeClassifier])

    Permalink

    uid

    stage uid

Type Members

  1. final type InputFeatures = (FeatureLike[RealNN], FeatureLike[OPVector])

    Permalink
    Definition Classes
    OpPipelineStage2 → OpPipelineStage → InputParams
  2. final type OutputFeatures = FeatureLike[Prediction]

    Permalink
    Definition Classes
    OpPipelineStage → OpPipelineStageBase

Value Members

  1. final def !=(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any
  2. final def ##(): Int

    Permalink
    Definition Classes
    AnyRef → Any
  3. final def $[T](param: Param[T]): T

    Permalink
    Attributes
    protected
    Definition Classes
    Params
  4. final def ==(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any
  5. final def asInstanceOf[T0]: T0

    Permalink
    Definition Classes
    Any
  6. final val cacheNodeIds: BooleanParam

    Permalink
    Definition Classes
    DecisionTreeParams
  7. final def checkInputLength(features: Array[_]): Boolean

    Permalink
    Definition Classes
    OpPipelineStage2 → InputParams
  8. def checkSerializable: Try[Unit]

    Permalink
    Definition Classes
    OpPipelineStageBase
  9. final val checkpointInterval: IntParam

    Permalink
    Definition Classes
    HasCheckpointInterval
  10. final def clear(param: Param[_]): OpDecisionTreeClassifier.this.type

    Permalink
    Definition Classes
    Params
  11. def clone(): AnyRef

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  12. final def copy(extra: ParamMap): OpDecisionTreeClassifier.this.type

    Permalink
    Definition Classes
    OpPipelineStageBase → Params
  13. def copyValues[T <: Params](to: T, extra: ParamMap): T

    Permalink
    Attributes
    protected
    Definition Classes
    Params
  14. final def defaultCopy[T <: Params](extra: ParamMap): T

    Permalink
    Attributes
    protected
    Definition Classes
    Params
  15. final def eq(arg0: AnyRef): Boolean

    Permalink
    Definition Classes
    AnyRef
  16. def equals(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any
  17. def explainParam(param: Param[_]): String

    Permalink
    Definition Classes
    Params
  18. def explainParams(): String

    Permalink
    Definition Classes
    Params
  19. final def extractParamMap(): ParamMap

    Permalink
    Definition Classes
    Params
  20. final def extractParamMap(extra: ParamMap): ParamMap

    Permalink
    Definition Classes
    Params
  21. final val featuresCol: Param[String]

    Permalink
    Definition Classes
    HasFeaturesCol
  22. def finalize(): Unit

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  23. def fit(dataset: Dataset[_]): OpPredictorWrapperModel[DecisionTreeClassificationModel]

    Permalink

    Function that fits the binary model

    Function that fits the binary model

    Definition Classes
    OpPredictorWrapper → Estimator
  24. def fit(dataset: Dataset[_], paramMaps: Array[ParamMap]): Seq[OpPredictorWrapperModel[DecisionTreeClassificationModel]]

    Permalink
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" )
  25. def fit(dataset: Dataset[_], paramMap: ParamMap): OpPredictorWrapperModel[DecisionTreeClassificationModel]

    Permalink
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" )
  26. def fit(dataset: Dataset[_], firstParamPair: ParamPair[_], otherParamPairs: ParamPair[_]*): OpPredictorWrapperModel[DecisionTreeClassificationModel]

    Permalink
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" ) @varargs()
  27. final def get[T](param: Param[T]): Option[T]

    Permalink
    Definition Classes
    Params
  28. final def getCacheNodeIds: Boolean

    Permalink
    Definition Classes
    DecisionTreeParams
  29. final def getCheckpointInterval: Int

    Permalink
    Definition Classes
    HasCheckpointInterval
  30. final def getClass(): Class[_]

    Permalink
    Definition Classes
    AnyRef → Any
  31. final def getDefault[T](param: Param[T]): Option[T]

    Permalink
    Definition Classes
    Params
  32. final def getFeaturesCol: String

    Permalink
    Definition Classes
    HasFeaturesCol
  33. final def getImpurity: String

    Permalink
    Definition Classes
    TreeClassifierParams
  34. final def getInputFeature[T <: FeatureType](i: Int): Option[FeatureLike[T]]

    Permalink
    Definition Classes
    InputParams
  35. final def getInputFeatures(): Array[OPFeature]

    Permalink
    Definition Classes
    InputParams
  36. final def getInputSchema(): StructType

    Permalink
    Definition Classes
    OpPipelineStageParams
  37. final def getLabelCol: String

    Permalink
    Definition Classes
    HasLabelCol
  38. final def getMaxBins: Int

    Permalink
    Definition Classes
    DecisionTreeParams
  39. final def getMaxDepth: Int

    Permalink
    Definition Classes
    DecisionTreeParams
  40. final def getMaxMemoryInMB: Int

    Permalink
    Definition Classes
    DecisionTreeParams
  41. final def getMetadata(): Metadata

    Permalink
    Definition Classes
    OpPipelineStageParams
  42. final def getMinInfoGain: Double

    Permalink
    Definition Classes
    DecisionTreeParams
  43. final def getMinInstancesPerNode: Int

    Permalink
    Definition Classes
    DecisionTreeParams
  44. final def getOrDefault[T](param: Param[T]): T

    Permalink
    Definition Classes
    Params
  45. def getOutput(): FeatureLike[Prediction]

    Permalink
    Definition Classes
    OpPipelineStage2 → OpPipelineStageBase
  46. final def getOutputFeatureName: String

    Permalink
    Definition Classes
    OpPipelineStage
  47. def getParam(paramName: String): Param[Any]

    Permalink
    Definition Classes
    Params
  48. final def getPredictionCol: String

    Permalink
    Definition Classes
    HasPredictionCol
  49. final def getSeed: Long

    Permalink
    Definition Classes
    HasSeed
  50. def getSparkMlStage(): Option[DecisionTreeClassifier]

    Permalink
    Definition Classes
    SparkWrapperParams
  51. def getStageSavePath(): Option[String]

    Permalink
    Definition Classes
    SparkWrapperParams
  52. final def getTransientFeature(i: Int): Option[TransientFeature]

    Permalink
    Definition Classes
    InputParams
  53. final def getTransientFeatures(): Array[TransientFeature]

    Permalink
    Definition Classes
    InputParams
  54. final def hasDefault[T](param: Param[T]): Boolean

    Permalink
    Definition Classes
    Params
  55. def hasParam(paramName: String): Boolean

    Permalink
    Definition Classes
    Params
  56. def hashCode(): Int

    Permalink
    Definition Classes
    AnyRef → Any
  57. final val impurity: Param[String]

    Permalink
    Definition Classes
    TreeClassifierParams
  58. final def in1: TransientFeature

    Permalink
    Attributes
    protected
    Definition Classes
    HasIn1
  59. final def in2: TransientFeature

    Permalink
    Attributes
    protected
    Definition Classes
    HasIn2
  60. def initializeLogIfNecessary(isInterpreter: Boolean): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  61. final def inputAsArray(in: InputFeatures): Array[OPFeature]

    Permalink
    Definition Classes
    OpPipelineStage2 → InputParams
  62. val inputParam1Name: String

    Permalink
    Definition Classes
    OpPredictorWrapper
  63. val inputParam2Name: String

    Permalink
    Definition Classes
    OpPredictorWrapper
  64. final def isDefined(param: Param[_]): Boolean

    Permalink
    Definition Classes
    Params
  65. final def isInstanceOf[T0]: Boolean

    Permalink
    Definition Classes
    Any
  66. final def isSet(param: Param[_]): Boolean

    Permalink
    Definition Classes
    Params
  67. def isTraceEnabled(): Boolean

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  68. final val labelCol: Param[String]

    Permalink
    Definition Classes
    HasLabelCol
  69. def log: Logger

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  70. def logDebug(msg: ⇒ String, throwable: Throwable): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  71. def logDebug(msg: ⇒ String): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  72. def logError(msg: ⇒ String, throwable: Throwable): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  73. def logError(msg: ⇒ String): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  74. def logInfo(msg: ⇒ String, throwable: Throwable): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  75. def logInfo(msg: ⇒ String): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  76. def logName: String

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  77. def logTrace(msg: ⇒ String, throwable: Throwable): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  78. def logTrace(msg: ⇒ String): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  79. def logWarning(msg: ⇒ String, throwable: Throwable): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  80. def logWarning(msg: ⇒ String): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  81. final val maxBins: IntParam

    Permalink
    Definition Classes
    DecisionTreeParams
  82. final val maxDepth: IntParam

    Permalink
    Definition Classes
    DecisionTreeParams
  83. final val maxMemoryInMB: IntParam

    Permalink
    Definition Classes
    DecisionTreeParams
  84. final val minInfoGain: DoubleParam

    Permalink
    Definition Classes
    DecisionTreeParams
  85. final val minInstancesPerNode: IntParam

    Permalink
    Definition Classes
    DecisionTreeParams
  86. final def ne(arg0: AnyRef): Boolean

    Permalink
    Definition Classes
    AnyRef
  87. final def notify(): Unit

    Permalink
    Definition Classes
    AnyRef
  88. final def notifyAll(): Unit

    Permalink
    Definition Classes
    AnyRef
  89. def onGetMetadata(): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    OpPipelineStageParams
  90. def onSetInput(): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    OpDecisionTreeClassifier → OpPipelineStageBase
  91. val operationName: String

    Permalink
    Definition Classes
    OpPredictorWrapper → OpPipelineStageBase
  92. final def outputAsArray(out: OutputFeatures): Array[OPFeature]

    Permalink
    Definition Classes
    OpPipelineStage → OpPipelineStageBase
  93. def outputFeatureUid: String

    Permalink
    Attributes
    protected[com.salesforce.op]
    Definition Classes
    OpPipelineStage2 → OpPipelineStage
  94. def outputIsResponse: Boolean

    Permalink
    Definition Classes
    OpPipelineStage
  95. val outputParamName: String

    Permalink
    Definition Classes
    OpPredictorWrapper
  96. lazy val params: Array[Param[_]]

    Permalink
    Definition Classes
    Params
  97. final val predictionCol: Param[String]

    Permalink
    Definition Classes
    HasPredictionCol
  98. val predictor: DecisionTreeClassifier

    Permalink

    the predictor to wrap

    the predictor to wrap

    Definition Classes
    OpPredictorWrapper
  99. def save(path: String): Unit

    Permalink
    Definition Classes
    MLWritable
    Annotations
    @Since( "1.6.0" ) @throws( ... )
  100. final val seed: LongParam

    Permalink
    Definition Classes
    HasSeed
  101. final def set(paramPair: ParamPair[_]): OpDecisionTreeClassifier.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    Params
  102. final def set(param: String, value: Any): OpDecisionTreeClassifier.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    Params
  103. final def set[T](param: Param[T], value: T): OpDecisionTreeClassifier.this.type

    Permalink
    Definition Classes
    Params
  104. def setCacheNodeIds(value: Boolean): OpDecisionTreeClassifier.this.type

    Permalink

    Definition Classes
    OpDecisionTreeClassifier → DecisionTreeParams
  105. def setCheckpointInterval(value: Int): OpDecisionTreeClassifier.this.type

    Permalink

    Specifies how often to checkpoint the cached node IDs.

    Specifies how often to checkpoint the cached node IDs. E.g. 10 means that the cache will get checkpointed every 10 iterations. This is only used if cacheNodeIds is true and if the checkpoint directory is set in org.apache.spark.SparkContext. Must be at least 1. (default = 10)

    Definition Classes
    OpDecisionTreeClassifier → DecisionTreeParams
  106. final def setDefault(paramPairs: ParamPair[_]*): OpDecisionTreeClassifier.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    Params
  107. final def setDefault[T](param: Param[T], value: T): OpDecisionTreeClassifier.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    Params
  108. def setImpurity(value: String): OpDecisionTreeClassifier.this.type

    Permalink

    Definition Classes
    OpDecisionTreeClassifier → TreeClassifierParams
  109. final def setInput(features: InputFeatures): OpDecisionTreeClassifier.this.type

    Permalink
    Definition Classes
    OpPipelineStageBase
  110. final def setInputFeatures[S <: OPFeature](features: Array[S]): OpDecisionTreeClassifier.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    InputParams
  111. def setMaxBins(value: Int): OpDecisionTreeClassifier.this.type

    Permalink

    Definition Classes
    OpDecisionTreeClassifier → DecisionTreeParams
  112. def setMaxDepth(value: Int): OpDecisionTreeClassifier.this.type

    Permalink

    Definition Classes
    OpDecisionTreeClassifier → DecisionTreeParams
  113. def setMaxMemoryInMB(value: Int): OpDecisionTreeClassifier.this.type

    Permalink

    Definition Classes
    OpDecisionTreeClassifier → DecisionTreeParams
  114. final def setMetadata(m: Metadata): OpDecisionTreeClassifier.this.type

    Permalink
    Definition Classes
    OpPipelineStageParams
  115. def setMinInfoGain(value: Double): OpDecisionTreeClassifier.this.type

    Permalink

    Definition Classes
    OpDecisionTreeClassifier → DecisionTreeParams
  116. def setMinInstancesPerNode(value: Int): OpDecisionTreeClassifier.this.type

    Permalink

    Definition Classes
    OpDecisionTreeClassifier → DecisionTreeParams
  117. def setOutputFeatureName(name: String): OpDecisionTreeClassifier.this.type

    Permalink
    Definition Classes
    OpPipelineStage
  118. def setSeed(value: Long): OpDecisionTreeClassifier.this.type

    Permalink

    Definition Classes
    OpDecisionTreeClassifier → DecisionTreeParams
  119. def setSparkMlStage(stage: Option[DecisionTreeClassifier]): OpDecisionTreeClassifier.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    SparkWrapperParams
  120. def setStageSavePath(path: String): OpDecisionTreeClassifier.this.type

    Permalink
    Definition Classes
    SparkWrapperParams
  121. final val sparkInputColParamNames: StringArrayParam

    Permalink
    Definition Classes
    SparkWrapperParams
  122. final val sparkMlStage: SparkStageParam[DecisionTreeClassifier]

    Permalink
    Definition Classes
    SparkWrapperParams
  123. final val sparkOutputColParamNames: StringArrayParam

    Permalink
    Definition Classes
    SparkWrapperParams
  124. final def stageName: String

    Permalink
    Definition Classes
    OpPipelineStageBase
  125. final def synchronized[T0](arg0: ⇒ T0): T0

    Permalink
    Definition Classes
    AnyRef
  126. def toString(): String

    Permalink
    Definition Classes
    Identifiable → AnyRef → Any
  127. final def transformSchema(schema: StructType): StructType

    Permalink
    Definition Classes
    OpPipelineStageBase
  128. def transformSchema(schema: StructType, logging: Boolean): StructType

    Permalink
    Attributes
    protected
    Definition Classes
    PipelineStage
    Annotations
    @DeveloperApi()
  129. implicit val tti1: scala.reflect.api.JavaUniverse.TypeTag[RealNN]

    Permalink
    Definition Classes
    OpPredictorWrapper
  130. implicit val tti2: scala.reflect.api.JavaUniverse.TypeTag[OPVector]

    Permalink
    Definition Classes
    OpPredictorWrapper
  131. implicit val tto: scala.reflect.api.JavaUniverse.TypeTag[Prediction]

    Permalink
    Definition Classes
    OpPredictorWrapper → OpPipelineStage2
  132. implicit val ttov: scala.reflect.api.JavaUniverse.TypeTag[Map[String, Double]]

    Permalink
    Definition Classes
    OpPredictorWrapper → OpPipelineStage2
  133. val uid: String

    Permalink

    stage uid

    stage uid

    Definition Classes
    OpPredictorWrapper → Identifiable
  134. def validateAndTransformSchema(schema: StructType, fitting: Boolean, featuresDataType: DataType): StructType

    Permalink
    Attributes
    protected
    Definition Classes
    PredictorParams
  135. final def wait(): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  136. final def wait(arg0: Long, arg1: Int): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  137. final def wait(arg0: Long): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  138. final def write: MLWriter

    Permalink
    Definition Classes
    OpPipelineStageBase → MLWritable

Inherited from DecisionTreeClassifierParams

Inherited from TreeClassifierParams

Inherited from DecisionTreeParams

Inherited from HasSeed

Inherited from HasCheckpointInterval

Inherited from PredictorParams

Inherited from HasPredictionCol

Inherited from HasFeaturesCol

Inherited from HasLabelCol

Inherited from OpPredictorWrapper[DecisionTreeClassifier, DecisionTreeClassificationModel]

Inherited from SparkWrapperParams[DecisionTreeClassifier]

Inherited from OpPipelineStage2[RealNN, OPVector, Prediction]

Inherited from HasIn2

Inherited from HasIn1

Inherited from OpPipelineStage[Prediction]

Inherited from OpPipelineStageBase

Inherited from MLWritable

Inherited from OpPipelineStageParams

Inherited from InputParams

Inherited from Estimator[OpPredictorWrapperModel[DecisionTreeClassificationModel]]

Inherited from PipelineStage

Inherited from Logging

Inherited from Params

Inherited from Serializable

Inherited from Serializable

Inherited from Identifiable

Inherited from AnyRef

Inherited from Any

expertSetParam

setParam

Ungrouped