Packages

trait AutoTrainer[TrainedModel <: Model[TrainedModel]] extends Estimator[TrainedModel] with HasLabelCol with HasInputCols with ComplexParamsWritable with HasFeaturesCol with Wrappable

Defines common inheritance and parameters across trainers.

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

Abstract Value Members

  1. abstract def copy(extra: ParamMap): Estimator[TrainedModel]
    Definition Classes
    Estimator → PipelineStage → Params
  2. abstract def fit(dataset: Dataset[_]): TrainedModel
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" )
  3. abstract def modelDoc: String

    Doc for model to run.

  4. abstract def transformSchema(schema: StructType): StructType
    Definition Classes
    PipelineStage
  5. abstract val uid: String
    Definition Classes
    Identifiable

Concrete 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[_]): AutoTrainer.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 copyValues[T <: Params](to: T, extra: ParamMap): T
    Attributes
    protected
    Definition Classes
    Params
  11. lazy val copyrightLines: String
    Attributes
    protected
    Definition Classes
    BaseWrappable
  12. final def defaultCopy[T <: Params](extra: ParamMap): T
    Attributes
    protected
    Definition Classes
    Params
  13. final def eq(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  14. def equals(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  15. def explainParam(param: Param[_]): String
    Definition Classes
    Params
  16. def explainParams(): String
    Definition Classes
    Params
  17. final def extractParamMap(): ParamMap
    Definition Classes
    Params
  18. final def extractParamMap(extra: ParamMap): ParamMap
    Definition Classes
    Params
  19. val featuresCol: Param[String]

    The name of the features column

    The name of the features column

    Definition Classes
    HasFeaturesCol
  20. def finalize(): Unit
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  21. def fit(dataset: Dataset[_], paramMaps: Seq[ParamMap]): Seq[TrainedModel]
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" )
  22. def fit(dataset: Dataset[_], paramMap: ParamMap): TrainedModel
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" )
  23. def fit(dataset: Dataset[_], firstParamPair: ParamPair[_], otherParamPairs: ParamPair[_]*): TrainedModel
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" ) @varargs()
  24. final def get[T](param: Param[T]): Option[T]
    Definition Classes
    Params
  25. final def getClass(): Class[_]
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  26. final def getDefault[T](param: Param[T]): Option[T]
    Definition Classes
    Params
  27. def getFeaturesCol: String

    Definition Classes
    HasFeaturesCol
  28. def getInputCols: Array[String]

    Definition Classes
    HasInputCols
  29. def getLabelCol: String

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

  31. def getNumFeatures: Int

  32. final def getOrDefault[T](param: Param[T]): T
    Definition Classes
    Params
  33. def getParam(paramName: String): Param[Any]
    Definition Classes
    Params
  34. def getParamInfo(p: Param[_]): ParamInfo[_]
    Definition Classes
    BaseWrappable
  35. final def hasDefault[T](param: Param[T]): Boolean
    Definition Classes
    Params
  36. def hasParam(paramName: String): Boolean
    Definition Classes
    Params
  37. def hashCode(): Int
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  38. def initializeLogIfNecessary(isInterpreter: Boolean, silent: Boolean): Boolean
    Attributes
    protected
    Definition Classes
    Logging
  39. def initializeLogIfNecessary(isInterpreter: Boolean): Unit
    Attributes
    protected
    Definition Classes
    Logging
  40. val inputCols: StringArrayParam

    The names of the inputColumns

    The names of the inputColumns

    Definition Classes
    HasInputCols
  41. final def isDefined(param: Param[_]): Boolean
    Definition Classes
    Params
  42. final def isInstanceOf[T0]: Boolean
    Definition Classes
    Any
  43. final def isSet(param: Param[_]): Boolean
    Definition Classes
    Params
  44. def isTraceEnabled(): Boolean
    Attributes
    protected
    Definition Classes
    Logging
  45. val labelCol: Param[String]

    The name of the label column

    The name of the label column

    Definition Classes
    HasLabelCol
  46. def log: Logger
    Attributes
    protected
    Definition Classes
    Logging
  47. def logDebug(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  48. def logDebug(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  49. def logError(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  50. def logError(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  51. def logInfo(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  52. def logInfo(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  53. def logName: String
    Attributes
    protected
    Definition Classes
    Logging
  54. def logTrace(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  55. def logTrace(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  56. def logWarning(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  57. def logWarning(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  58. def makePyFile(conf: CodegenConfig): Unit
    Definition Classes
    PythonWrappable
  59. def makeRFile(conf: CodegenConfig): Unit
    Definition Classes
    RWrappable
  60. val model: EstimatorParam

    Model to run.

    Model to run. See doc on derived classes.

  61. final def ne(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  62. final def notify(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  63. final def notifyAll(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  64. val numFeatures: IntParam

    Number of features to hash to

  65. lazy val params: Array[Param[_]]
    Definition Classes
    Params
  66. def pyAdditionalMethods: String
    Definition Classes
    PythonWrappable
  67. lazy val pyClassDoc: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  68. lazy val pyClassName: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  69. def pyExtraEstimatorImports: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  70. def pyExtraEstimatorMethods: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  71. lazy val pyInheritedClasses: Seq[String]
    Attributes
    protected
    Definition Classes
    PythonWrappable
  72. def pyInitFunc(): String
    Definition Classes
    PythonWrappable
  73. lazy val pyInternalWrapper: Boolean
    Attributes
    protected
    Definition Classes
    PythonWrappable
  74. lazy val pyObjectBaseClass: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  75. def pyParamArg[T](p: Param[T]): String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  76. def pyParamDefault[T](p: Param[T]): Option[String]
    Attributes
    protected
    Definition Classes
    PythonWrappable
  77. def pyParamGetter(p: Param[_]): String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  78. def pyParamSetter(p: Param[_]): String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  79. def pyParamsArgs: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  80. def pyParamsDefaults: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  81. lazy val pyParamsDefinitions: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  82. def pyParamsGetters: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  83. def pyParamsSetters: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  84. def pythonClass(): String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  85. def rClass(): String
    Attributes
    protected
    Definition Classes
    RWrappable
  86. def rDocString: String
    Attributes
    protected
    Definition Classes
    RWrappable
  87. def rExtraBodyLines: String
    Attributes
    protected
    Definition Classes
    RWrappable
  88. def rExtraInitLines: String
    Attributes
    protected
    Definition Classes
    RWrappable
  89. lazy val rFuncName: String
    Attributes
    protected
    Definition Classes
    RWrappable
  90. lazy val rInternalWrapper: Boolean
    Attributes
    protected
    Definition Classes
    RWrappable
  91. def rParamArg[T](p: Param[T]): String
    Attributes
    protected
    Definition Classes
    RWrappable
  92. def rParamsArgs: String
    Attributes
    protected
    Definition Classes
    RWrappable
  93. def rSetterLines: String
    Attributes
    protected
    Definition Classes
    RWrappable
  94. def save(path: String): Unit
    Definition Classes
    MLWritable
    Annotations
    @Since( "1.6.0" ) @throws( ... )
  95. final def set(paramPair: ParamPair[_]): AutoTrainer.this.type
    Attributes
    protected
    Definition Classes
    Params
  96. final def set(param: String, value: Any): AutoTrainer.this.type
    Attributes
    protected
    Definition Classes
    Params
  97. final def set[T](param: Param[T], value: T): AutoTrainer.this.type
    Definition Classes
    Params
  98. final def setDefault(paramPairs: ParamPair[_]*): AutoTrainer.this.type
    Attributes
    protected
    Definition Classes
    Params
  99. final def setDefault[T](param: Param[T], value: T): AutoTrainer.this.type
    Attributes
    protected[org.apache.spark.ml]
    Definition Classes
    Params
  100. def setFeaturesCol(value: String): AutoTrainer.this.type

    Definition Classes
    HasFeaturesCol
  101. def setInputCols(value: Array[String]): AutoTrainer.this.type

    Definition Classes
    HasInputCols
  102. def setLabelCol(value: String): AutoTrainer.this.type

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

  104. def setNumFeatures(value: Int): AutoTrainer.this.type

  105. final def synchronized[T0](arg0: ⇒ T0): T0
    Definition Classes
    AnyRef
  106. val thisStage: Params
    Attributes
    protected
    Definition Classes
    BaseWrappable
  107. def toString(): String
    Definition Classes
    Identifiable → AnyRef → Any
  108. def transformSchema(schema: StructType, logging: Boolean): StructType
    Attributes
    protected
    Definition Classes
    PipelineStage
    Annotations
    @DeveloperApi()
  109. final def wait(): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  110. final def wait(arg0: Long, arg1: Int): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  111. final def wait(arg0: Long): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()
  112. def write: MLWriter
    Definition Classes
    ComplexParamsWritable → MLWritable

Inherited from Wrappable

Inherited from RWrappable

Inherited from PythonWrappable

Inherited from BaseWrappable

Inherited from HasFeaturesCol

Inherited from ComplexParamsWritable

Inherited from MLWritable

Inherited from HasInputCols

Inherited from HasLabelCol

Inherited from Estimator[TrainedModel]

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