Packages

class TrainRegressor extends Estimator[TrainedRegressorModel] with AutoTrainer[TrainedRegressorModel] with BasicLogging

Trains a regression model.

Linear Supertypes
BasicLogging, AutoTrainer[TrainedRegressorModel], Wrappable, DotnetWrappable, RWrappable, PythonWrappable, BaseWrappable, HasFeaturesCol, ComplexParamsWritable, MLWritable, HasLabelCol, Estimator[TrainedRegressorModel], PipelineStage, Logging, Params, Serializable, Serializable, Identifiable, AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. TrainRegressor
  2. BasicLogging
  3. AutoTrainer
  4. Wrappable
  5. DotnetWrappable
  6. RWrappable
  7. PythonWrappable
  8. BaseWrappable
  9. HasFeaturesCol
  10. ComplexParamsWritable
  11. MLWritable
  12. HasLabelCol
  13. Estimator
  14. PipelineStage
  15. Logging
  16. Params
  17. Serializable
  18. Serializable
  19. Identifiable
  20. AnyRef
  21. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new TrainRegressor()
  2. new TrainRegressor(uid: String)

Value Members

  1. final def !=(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  2. final def ##(): Int
    Definition Classes
    AnyRef → Any
  3. final def $[T](param: Param[T]): T
    Attributes
    protected
    Definition Classes
    Params
  4. final def ==(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  5. final def asInstanceOf[T0]: T0
    Definition Classes
    Any
  6. lazy val classNameHelper: String
    Attributes
    protected
    Definition Classes
    BaseWrappable
  7. final def clear(param: Param[_]): TrainRegressor.this.type
    Definition Classes
    Params
  8. def clone(): AnyRef
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()
  9. def companionModelClassName: String
    Attributes
    protected
    Definition Classes
    BaseWrappable
  10. def copy(extra: ParamMap): Estimator[TrainedRegressorModel]
    Definition Classes
    TrainRegressor → Estimator → PipelineStage → Params
  11. def copyValues[T <: Params](to: T, extra: ParamMap): T
    Attributes
    protected
    Definition Classes
    Params
  12. lazy val copyrightLines: String
    Attributes
    protected
    Definition Classes
    BaseWrappable
  13. final def defaultCopy[T <: Params](extra: ParamMap): T
    Attributes
    protected
    Definition Classes
    Params
  14. def dotnetAdditionalMethods: String
    Definition Classes
    DotnetWrappable
  15. def dotnetClass(): String
    Attributes
    protected
    Definition Classes
    DotnetWrappable
  16. lazy val dotnetClassName: String
    Attributes
    protected
    Definition Classes
    DotnetWrappable
  17. lazy val dotnetClassNameString: String
    Attributes
    protected
    Definition Classes
    DotnetWrappable
  18. lazy val dotnetClassWrapperName: String
    Attributes
    protected
    Definition Classes
    DotnetWrappable
  19. lazy val dotnetCopyrightLines: String
    Attributes
    protected
    Definition Classes
    DotnetWrappable
  20. def dotnetExtraEstimatorImports: String
    Attributes
    protected
    Definition Classes
    DotnetWrappable
  21. def dotnetExtraMethods: String
    Attributes
    protected
    Definition Classes
    DotnetWrappable
  22. lazy val dotnetInternalWrapper: Boolean
    Attributes
    protected
    Definition Classes
    DotnetWrappable
  23. def dotnetMLReadWriteMethods: String
    Attributes
    protected
    Definition Classes
    DotnetWrappable
  24. lazy val dotnetNamespace: String
    Attributes
    protected
    Definition Classes
    DotnetWrappable
  25. lazy val dotnetObjectBaseClass: String
    Attributes
    protected
    Definition Classes
    DotnetWrappable
  26. def dotnetParamGetter(p: Param[_]): String
    Attributes
    protected
    Definition Classes
    DotnetWrappable
  27. def dotnetParamGetters: String
    Attributes
    protected
    Definition Classes
    DotnetWrappable
  28. def dotnetParamSetter(p: Param[_]): String
    Attributes
    protected
    Definition Classes
    DotnetWrappable
  29. def dotnetParamSetters: String
    Attributes
    protected
    Definition Classes
    DotnetWrappable
  30. def dotnetWrapAsTypeMethod: String
    Attributes
    protected
    Definition Classes
    DotnetWrappable
  31. final def eq(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  32. def equals(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  33. def explainParam(param: Param[_]): String
    Definition Classes
    Params
  34. def explainParams(): String
    Definition Classes
    Params
  35. final def extractParamMap(): ParamMap
    Definition Classes
    Params
  36. final def extractParamMap(extra: ParamMap): ParamMap
    Definition Classes
    Params
  37. val featuresCol: Param[String]

    The name of the features column

    The name of the features column

    Definition Classes
    HasFeaturesCol
  38. def finalize(): Unit
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  39. def fit(dataset: Dataset[_]): TrainedRegressorModel

    Fits the regression model.

    Fits the regression model.

    dataset

    The input dataset to train.

    returns

    The trained regression model.

    Definition Classes
    TrainRegressor → Estimator
  40. def fit(dataset: Dataset[_], paramMaps: Seq[ParamMap]): Seq[TrainedRegressorModel]
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" )
  41. def fit(dataset: Dataset[_], paramMap: ParamMap): TrainedRegressorModel
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" )
  42. def fit(dataset: Dataset[_], firstParamPair: ParamPair[_], otherParamPairs: ParamPair[_]*): TrainedRegressorModel
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" ) @varargs()
  43. final def get[T](param: Param[T]): Option[T]
    Definition Classes
    Params
  44. final def getClass(): Class[_]
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  45. final def getDefault[T](param: Param[T]): Option[T]
    Definition Classes
    Params
  46. def getFeaturesCol: String

    Definition Classes
    HasFeaturesCol
  47. def getLabelCol: String

    Definition Classes
    HasLabelCol
  48. def getModel: Estimator[_ <: Model[_]]

    Definition Classes
    AutoTrainer
  49. def getNumFeatures: Int

    Definition Classes
    AutoTrainer
  50. final def getOrDefault[T](param: Param[T]): T
    Definition Classes
    Params
  51. def getParam(paramName: String): Param[Any]
    Definition Classes
    Params
  52. def getParamInfo(p: Param[_]): ParamInfo[_]
    Definition Classes
    BaseWrappable
  53. final def hasDefault[T](param: Param[T]): Boolean
    Definition Classes
    Params
  54. def hasParam(paramName: String): Boolean
    Definition Classes
    Params
  55. def hashCode(): Int
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  56. def initializeLogIfNecessary(isInterpreter: Boolean, silent: Boolean): Boolean
    Attributes
    protected
    Definition Classes
    Logging
  57. def initializeLogIfNecessary(isInterpreter: Boolean): Unit
    Attributes
    protected
    Definition Classes
    Logging
  58. final def isDefined(param: Param[_]): Boolean
    Definition Classes
    Params
  59. final def isInstanceOf[T0]: Boolean
    Definition Classes
    Any
  60. final def isSet(param: Param[_]): Boolean
    Definition Classes
    Params
  61. def isTraceEnabled(): Boolean
    Attributes
    protected
    Definition Classes
    Logging
  62. val labelCol: Param[String]

    The name of the label column

    The name of the label column

    Definition Classes
    HasLabelCol
  63. def log: Logger
    Attributes
    protected
    Definition Classes
    Logging
  64. def logBase(methodName: String): Unit
    Attributes
    protected
    Definition Classes
    BasicLogging
  65. def logClass(): Unit
    Definition Classes
    BasicLogging
  66. def logDebug(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  67. def logDebug(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  68. def logError(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  69. def logError(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  70. def logErrorBase(methodName: String, e: Exception): Unit
    Attributes
    protected
    Definition Classes
    BasicLogging
  71. def logFit[T](f: ⇒ T): T
    Definition Classes
    BasicLogging
  72. def logInfo(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  73. def logInfo(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  74. def logName: String
    Attributes
    protected
    Definition Classes
    Logging
  75. def logPredict[T](f: ⇒ T): T
    Definition Classes
    BasicLogging
  76. def logTrace(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  77. def logTrace(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  78. def logTrain[T](f: ⇒ T): T
    Definition Classes
    BasicLogging
  79. def logTransform[T](f: ⇒ T): T
    Definition Classes
    BasicLogging
  80. def logVerb[T](verb: String, f: ⇒ T): T
    Definition Classes
    BasicLogging
  81. def logWarning(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  82. def logWarning(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  83. def makeDotnetFile(conf: CodegenConfig): Unit
    Definition Classes
    DotnetWrappable
  84. def makePyFile(conf: CodegenConfig): Unit
    Definition Classes
    PythonWrappable
  85. def makeRFile(conf: CodegenConfig): Unit
    Definition Classes
    RWrappable
  86. val model: EstimatorParam

    Model to run.

    Model to run. See doc on derived classes.

    Definition Classes
    AutoTrainer
  87. def modelDoc: String

    Doc for model to run.

    Doc for model to run.

    Definition Classes
    TrainRegressor → AutoTrainer
  88. final def ne(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  89. final def notify(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  90. final def notifyAll(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  91. val numFeatures: IntParam

    Number of features to hash to

    Number of features to hash to

    Definition Classes
    AutoTrainer
  92. lazy val params: Array[Param[_]]
    Definition Classes
    Params
  93. def pyAdditionalMethods: String
    Definition Classes
    PythonWrappable
  94. lazy val pyClassDoc: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  95. lazy val pyClassName: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  96. def pyExtraEstimatorImports: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  97. def pyExtraEstimatorMethods: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  98. lazy val pyInheritedClasses: Seq[String]
    Attributes
    protected
    Definition Classes
    PythonWrappable
  99. def pyInitFunc(): String
    Definition Classes
    PythonWrappable
  100. lazy val pyInternalWrapper: Boolean
    Attributes
    protected
    Definition Classes
    PythonWrappable
  101. lazy val pyObjectBaseClass: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  102. def pyParamArg[T](p: Param[T]): String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  103. def pyParamDefault[T](p: Param[T]): Option[String]
    Attributes
    protected
    Definition Classes
    PythonWrappable
  104. def pyParamGetter(p: Param[_]): String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  105. def pyParamSetter(p: Param[_]): String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  106. def pyParamsArgs: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  107. def pyParamsDefaults: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  108. lazy val pyParamsDefinitions: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  109. def pyParamsGetters: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  110. def pyParamsSetters: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  111. def pythonClass(): String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  112. def rClass(): String
    Attributes
    protected
    Definition Classes
    RWrappable
  113. def rDocString: String
    Attributes
    protected
    Definition Classes
    RWrappable
  114. def rExtraBodyLines: String
    Attributes
    protected
    Definition Classes
    RWrappable
  115. def rExtraInitLines: String
    Attributes
    protected
    Definition Classes
    RWrappable
  116. lazy val rFuncName: String
    Attributes
    protected
    Definition Classes
    RWrappable
  117. lazy val rInternalWrapper: Boolean
    Attributes
    protected
    Definition Classes
    RWrappable
  118. def rParamArg[T](p: Param[T]): String
    Attributes
    protected
    Definition Classes
    RWrappable
  119. def rParamsArgs: String
    Attributes
    protected
    Definition Classes
    RWrappable
  120. def rSetterLines: String
    Attributes
    protected
    Definition Classes
    RWrappable
  121. def save(path: String): Unit
    Definition Classes
    MLWritable
    Annotations
    @Since( "1.6.0" ) @throws( ... )
  122. final def set(paramPair: ParamPair[_]): TrainRegressor.this.type
    Attributes
    protected
    Definition Classes
    Params
  123. final def set(param: String, value: Any): TrainRegressor.this.type
    Attributes
    protected
    Definition Classes
    Params
  124. final def set[T](param: Param[T], value: T): TrainRegressor.this.type
    Definition Classes
    Params
  125. final def setDefault(paramPairs: ParamPair[_]*): TrainRegressor.this.type
    Attributes
    protected
    Definition Classes
    Params
  126. final def setDefault[T](param: Param[T], value: T): TrainRegressor.this.type
    Attributes
    protected
    Definition Classes
    Params
  127. def setFeaturesCol(value: String): TrainRegressor.this.type

    Definition Classes
    HasFeaturesCol
  128. def setLabelCol(value: String): TrainRegressor.this.type

    Definition Classes
    HasLabelCol
  129. def setModel(value: Estimator[_ <: Model[_]]): TrainRegressor.this.type

    Definition Classes
    AutoTrainer
  130. def setNumFeatures(value: Int): TrainRegressor.this.type

    Definition Classes
    AutoTrainer
  131. final def synchronized[T0](arg0: ⇒ T0): T0
    Definition Classes
    AnyRef
  132. val thisStage: Params
    Attributes
    protected
    Definition Classes
    BaseWrappable
  133. def toString(): String
    Definition Classes
    Identifiable → AnyRef → Any
  134. def transformSchema(schema: StructType): StructType
    Definition Classes
    TrainRegressor → PipelineStage
    Annotations
    @DeveloperApi()
  135. def transformSchema(schema: StructType, logging: Boolean): StructType
    Attributes
    protected
    Definition Classes
    PipelineStage
    Annotations
    @DeveloperApi()
  136. val uid: String
    Definition Classes
    TrainRegressor → BasicLogging → Identifiable
  137. val ver: String
    Definition Classes
    BasicLogging
  138. final def wait(): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  139. final def wait(arg0: Long, arg1: Int): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  140. final def wait(arg0: Long): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()
  141. def write: MLWriter
    Definition Classes
    ComplexParamsWritable → MLWritable

Inherited from BasicLogging

Inherited from AutoTrainer[TrainedRegressorModel]

Inherited from Wrappable

Inherited from DotnetWrappable

Inherited from RWrappable

Inherited from PythonWrappable

Inherited from BaseWrappable

Inherited from HasFeaturesCol

Inherited from ComplexParamsWritable

Inherited from MLWritable

Inherited from HasLabelCol

Inherited from Estimator[TrainedRegressorModel]

Inherited from PipelineStage

Inherited from Logging

Inherited from Params

Inherited from Serializable

Inherited from Serializable

Inherited from Identifiable

Inherited from AnyRef

Inherited from Any

getParam

param

setParam

Ungrouped