class VowpalWabbitRegressor extends BaseRegressor[Row, VowpalWabbitRegressor, VowpalWabbitRegressionModel] with VowpalWabbitBase with ComplexParamsWritable with BasicLogging

Linear Supertypes
BasicLogging, ComplexParamsWritable, MLWritable, VowpalWabbitBase, HasAdditionalFeatures, HasWeightCol, Wrappable, RWrappable, PythonWrappable, BaseWrappable, BaseRegressor[Row, VowpalWabbitRegressor, VowpalWabbitRegressionModel], Regressor[Row, VowpalWabbitRegressor, VowpalWabbitRegressionModel], Predictor[Row, VowpalWabbitRegressor, VowpalWabbitRegressionModel], PredictorParams, HasPredictionCol, HasFeaturesCol, HasLabelCol, Estimator[VowpalWabbitRegressionModel], PipelineStage, Logging, Params, Serializable, Serializable, Identifiable, AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. VowpalWabbitRegressor
  2. BasicLogging
  3. ComplexParamsWritable
  4. MLWritable
  5. VowpalWabbitBase
  6. HasAdditionalFeatures
  7. HasWeightCol
  8. Wrappable
  9. RWrappable
  10. PythonWrappable
  11. BaseWrappable
  12. BaseRegressor
  13. Regressor
  14. Predictor
  15. PredictorParams
  16. HasPredictionCol
  17. HasFeaturesCol
  18. HasLabelCol
  19. Estimator
  20. PipelineStage
  21. Logging
  22. Params
  23. Serializable
  24. Serializable
  25. Identifiable
  26. AnyRef
  27. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

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

Type Members

  1. implicit class ParamStringBuilder extends AnyRef
    Definition Classes
    VowpalWabbitBase
  2. class TrainContext extends AnyRef
    Definition Classes
    VowpalWabbitBase

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. def addExtraArgs(args: StringBuilder): Unit

    * Allow subclasses to add further arguments

    * Allow subclasses to add further arguments

    args

    argument builder to append to

    Attributes
    protected
    Definition Classes
    VowpalWabbitBase
  6. val additionalFeatures: StringArrayParam
    Definition Classes
    HasAdditionalFeatures
  7. val args: Param[String]
    Definition Classes
    VowpalWabbitBase
  8. final def asInstanceOf[T0]: T0
    Definition Classes
    Any
  9. lazy val classNameHelper: String
    Attributes
    protected
    Definition Classes
    BaseWrappable
  10. final def clear(param: Param[_]): VowpalWabbitRegressor.this.type
    Definition Classes
    Params
  11. def clone(): AnyRef
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()
  12. def companionModelClassName: String
    Attributes
    protected
    Definition Classes
    BaseWrappable
  13. def copy(extra: ParamMap): VowpalWabbitRegressor
    Definition Classes
    VowpalWabbitRegressor → Predictor → Estimator → PipelineStage → Params
  14. def copyValues[T <: Params](to: T, extra: ParamMap): T
    Attributes
    protected
    Definition Classes
    Params
  15. lazy val copyrightLines: String
    Attributes
    protected
    Definition Classes
    BaseWrappable
  16. def createLabelSetter(schema: StructType): (Row, VowpalWabbitExample) ⇒ Unit
    Attributes
    protected
    Definition Classes
    VowpalWabbitBase
  17. final def defaultCopy[T <: Params](extra: ParamMap): T
    Attributes
    protected
    Definition Classes
    Params
  18. final def eq(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  19. def equals(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  20. def explainParam(param: Param[_]): String
    Definition Classes
    Params
  21. def explainParams(): String
    Definition Classes
    Params
  22. def extractInstances(dataset: Dataset[_], validateInstance: (Instance) ⇒ Unit): RDD[Instance]
    Attributes
    protected
    Definition Classes
    PredictorParams
  23. def extractInstances(dataset: Dataset[_]): RDD[Instance]
    Attributes
    protected
    Definition Classes
    PredictorParams
  24. def extractLabeledPoints(dataset: Dataset[_]): RDD[LabeledPoint]
    Attributes
    protected
    Definition Classes
    Predictor
  25. final def extractParamMap(): ParamMap
    Definition Classes
    Params
  26. final def extractParamMap(extra: ParamMap): ParamMap
    Definition Classes
    Params
  27. final val featuresCol: Param[String]
    Definition Classes
    HasFeaturesCol
  28. def finalize(): Unit
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  29. def fit(dataset: Dataset[_]): VowpalWabbitRegressionModel
    Definition Classes
    Predictor → Estimator
  30. def fit(dataset: Dataset[_], paramMaps: Seq[ParamMap]): Seq[VowpalWabbitRegressionModel]
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" )
  31. def fit(dataset: Dataset[_], paramMap: ParamMap): VowpalWabbitRegressionModel
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" )
  32. def fit(dataset: Dataset[_], firstParamPair: ParamPair[_], otherParamPairs: ParamPair[_]*): VowpalWabbitRegressionModel
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" ) @varargs()
  33. final def get[T](param: Param[T]): Option[T]
    Definition Classes
    Params
  34. def getAdditionalColumns: Seq[String]
    Attributes
    protected
    Definition Classes
    VowpalWabbitBase
  35. def getAdditionalFeatures: Array[String]
    Definition Classes
    HasAdditionalFeatures
  36. def getArgs: String
    Definition Classes
    VowpalWabbitBase
  37. def getAsFloat(schema: StructType, idx: Int): (Row) ⇒ Float
    Attributes
    protected
    Definition Classes
    VowpalWabbitBase
  38. final def getClass(): Class[_]
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  39. final def getDefault[T](param: Param[T]): Option[T]
    Definition Classes
    Params
  40. final def getFeaturesCol: String
    Definition Classes
    HasFeaturesCol
  41. def getHashSeed: Int
    Definition Classes
    VowpalWabbitBase
  42. def getIgnoreNamespaces: String
    Definition Classes
    VowpalWabbitBase
  43. def getInitialModel: Array[Byte]
    Definition Classes
    VowpalWabbitBase
  44. def getInteractions: Array[String]
    Definition Classes
    VowpalWabbitBase
  45. def getL1: Double
    Definition Classes
    VowpalWabbitBase
  46. def getL2: Double
    Definition Classes
    VowpalWabbitBase
  47. final def getLabelCol: String
    Definition Classes
    HasLabelCol
  48. def getLearningRate: Double
    Definition Classes
    VowpalWabbitBase
  49. def getNumBits: Int
    Definition Classes
    VowpalWabbitBase
  50. def getNumPasses: Int
    Definition Classes
    VowpalWabbitBase
  51. final def getOrDefault[T](param: Param[T]): T
    Definition Classes
    Params
  52. def getParam(paramName: String): Param[Any]
    Definition Classes
    Params
  53. def getPowerT: Double
    Definition Classes
    VowpalWabbitBase
  54. final def getPredictionCol: String
    Definition Classes
    HasPredictionCol
  55. def getUseBarrierExecutionMode: Boolean
    Definition Classes
    VowpalWabbitBase
  56. def getWeightCol: String

    Definition Classes
    HasWeightCol
  57. final def hasDefault[T](param: Param[T]): Boolean
    Definition Classes
    Params
  58. def hasParam(paramName: String): Boolean
    Definition Classes
    Params
  59. def hashCode(): Int
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  60. val hashSeed: IntParam
    Definition Classes
    VowpalWabbitBase
  61. val ignoreNamespaces: Param[String]
    Definition Classes
    VowpalWabbitBase
  62. val initialModel: ByteArrayParam
    Definition Classes
    VowpalWabbitBase
  63. def initializeLogIfNecessary(isInterpreter: Boolean, silent: Boolean): Boolean
    Attributes
    protected
    Definition Classes
    Logging
  64. def initializeLogIfNecessary(isInterpreter: Boolean): Unit
    Attributes
    protected
    Definition Classes
    Logging
  65. val interactions: StringArrayParam
    Definition Classes
    VowpalWabbitBase
  66. final def isDefined(param: Param[_]): Boolean
    Definition Classes
    Params
  67. final def isInstanceOf[T0]: Boolean
    Definition Classes
    Any
  68. final def isSet(param: Param[_]): Boolean
    Definition Classes
    Params
  69. def isTraceEnabled(): Boolean
    Attributes
    protected
    Definition Classes
    Logging
  70. val l1: DoubleParam
    Definition Classes
    VowpalWabbitBase
  71. val l2: DoubleParam
    Definition Classes
    VowpalWabbitBase
  72. final val labelCol: Param[String]
    Definition Classes
    HasLabelCol
  73. val learningRate: DoubleParam
    Definition Classes
    VowpalWabbitBase
  74. def log: Logger
    Attributes
    protected
    Definition Classes
    Logging
  75. def logBase(methodName: String): Unit
    Attributes
    protected
    Definition Classes
    BasicLogging
  76. def logClass(): Unit
    Definition Classes
    BasicLogging
  77. def logDebug(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  78. def logDebug(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  79. def logError(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  80. def logError(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  81. def logErrorBase(methodName: String, e: Exception): Unit
    Attributes
    protected
    Definition Classes
    BasicLogging
  82. def logFit[T](f: ⇒ T): T
    Definition Classes
    BasicLogging
  83. def logInfo(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  84. def logInfo(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  85. def logName: String
    Attributes
    protected
    Definition Classes
    Logging
  86. def logPredict[T](f: ⇒ T): T
    Definition Classes
    BasicLogging
  87. def logTrace(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  88. def logTrace(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  89. def logTrain[T](f: ⇒ T): T
    Definition Classes
    BasicLogging
  90. def logTransform[T](f: ⇒ T): T
    Definition Classes
    BasicLogging
  91. def logVerb[T](verb: String, f: ⇒ T): T
    Definition Classes
    BasicLogging
  92. def logWarning(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  93. def logWarning(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  94. def makePyFile(conf: CodegenConfig): Unit
    Definition Classes
    PythonWrappable
  95. def makeRFile(conf: CodegenConfig): Unit
    Definition Classes
    RWrappable
  96. final def ne(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  97. final def notify(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  98. final def notifyAll(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  99. val numBits: IntParam
    Definition Classes
    VowpalWabbitBase
  100. val numPasses: IntParam
    Definition Classes
    VowpalWabbitBase
  101. lazy val params: Array[Param[_]]
    Definition Classes
    Params
  102. val powerT: DoubleParam
    Definition Classes
    VowpalWabbitBase
  103. final val predictionCol: Param[String]
    Definition Classes
    HasPredictionCol
  104. def pyAdditionalMethods: String
    Definition Classes
    PythonWrappable
  105. lazy val pyClassDoc: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  106. lazy val pyClassName: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  107. def pyExtraEstimatorImports: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  108. def pyExtraEstimatorMethods: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  109. lazy val pyInheritedClasses: Seq[String]
    Attributes
    protected
    Definition Classes
    PythonWrappable
  110. def pyInitFunc(): String
    Definition Classes
    PythonWrappable
  111. lazy val pyInternalWrapper: Boolean
    Attributes
    protected
    Definition Classes
    VowpalWabbitRegressor → VowpalWabbitBase → PythonWrappable
  112. lazy val pyObjectBaseClass: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  113. def pyParamArg[T](p: Param[T]): String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  114. def pyParamDefault[T](p: Param[T]): Option[String]
    Attributes
    protected
    Definition Classes
    PythonWrappable
  115. def pyParamGetter(p: Param[_]): String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  116. def pyParamSetter(p: Param[_]): String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  117. def pyParamsArgs: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  118. def pyParamsDefaults: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  119. lazy val pyParamsDefinitions: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  120. def pyParamsGetters: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  121. def pyParamsSetters: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  122. def pythonClass(): String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  123. def rClass(): String
    Attributes
    protected
    Definition Classes
    RWrappable
  124. def rDocString: String
    Attributes
    protected
    Definition Classes
    RWrappable
  125. def rExtraBodyLines: String
    Attributes
    protected
    Definition Classes
    RWrappable
  126. def rExtraInitLines: String
    Attributes
    protected
    Definition Classes
    RWrappable
  127. lazy val rFuncName: String
    Attributes
    protected
    Definition Classes
    RWrappable
  128. lazy val rInternalWrapper: Boolean
    Attributes
    protected
    Definition Classes
    RWrappable
  129. def rParamArg[T](p: Param[T]): String
    Attributes
    protected
    Definition Classes
    RWrappable
  130. def rParamsArgs: String
    Attributes
    protected
    Definition Classes
    RWrappable
  131. def rSetterLines: String
    Attributes
    protected
    Definition Classes
    RWrappable
  132. def save(path: String): Unit
    Definition Classes
    MLWritable
    Annotations
    @Since( "1.6.0" ) @throws( ... )
  133. final def set(paramPair: ParamPair[_]): VowpalWabbitRegressor.this.type
    Attributes
    protected
    Definition Classes
    Params
  134. final def set(param: String, value: Any): VowpalWabbitRegressor.this.type
    Attributes
    protected
    Definition Classes
    Params
  135. final def set[T](param: Param[T], value: T): VowpalWabbitRegressor.this.type
    Definition Classes
    Params
  136. def setAdditionalFeatures(value: Array[String]): VowpalWabbitRegressor.this.type
    Definition Classes
    HasAdditionalFeatures
  137. def setArgs(value: String): VowpalWabbitRegressor.this.type
    Definition Classes
    VowpalWabbitBase
  138. final def setDefault(paramPairs: ParamPair[_]*): VowpalWabbitRegressor.this.type
    Attributes
    protected
    Definition Classes
    Params
  139. final def setDefault[T](param: Param[T], value: T): VowpalWabbitRegressor.this.type
    Attributes
    protected
    Definition Classes
    Params
  140. def setFeaturesCol(value: String): VowpalWabbitRegressor
    Definition Classes
    Predictor
  141. def setHashSeed(value: Int): VowpalWabbitRegressor.this.type
    Definition Classes
    VowpalWabbitBase
  142. def setIgnoreNamespaces(value: String): VowpalWabbitRegressor.this.type
    Definition Classes
    VowpalWabbitBase
  143. def setInitialModel(value: Array[Byte]): VowpalWabbitRegressor.this.type
    Definition Classes
    VowpalWabbitBase
  144. def setInteractions(value: Array[String]): VowpalWabbitRegressor.this.type
    Definition Classes
    VowpalWabbitBase
  145. def setL1(value: Double): VowpalWabbitRegressor.this.type
    Definition Classes
    VowpalWabbitBase
  146. def setL2(value: Double): VowpalWabbitRegressor.this.type
    Definition Classes
    VowpalWabbitBase
  147. def setLabelCol(value: String): VowpalWabbitRegressor
    Definition Classes
    Predictor
  148. def setLearningRate(value: Double): VowpalWabbitRegressor.this.type
    Definition Classes
    VowpalWabbitBase
  149. def setNumBits(value: Int): VowpalWabbitRegressor.this.type
    Definition Classes
    VowpalWabbitBase
  150. def setNumPasses(value: Int): VowpalWabbitRegressor.this.type
    Definition Classes
    VowpalWabbitBase
  151. def setPowerT(value: Double): VowpalWabbitRegressor.this.type
    Definition Classes
    VowpalWabbitBase
  152. def setPredictionCol(value: String): VowpalWabbitRegressor
    Definition Classes
    Predictor
  153. def setUseBarrierExecutionMode(value: Boolean): VowpalWabbitRegressor.this.type
    Definition Classes
    VowpalWabbitBase
  154. def setWeightCol(value: String): VowpalWabbitRegressor.this.type

    Definition Classes
    HasWeightCol
  155. final def synchronized[T0](arg0: ⇒ T0): T0
    Definition Classes
    AnyRef
  156. def toString(): String
    Definition Classes
    Identifiable → AnyRef → Any
  157. def train(dataset: Dataset[_]): VowpalWabbitRegressionModel
    Definition Classes
    VowpalWabbitRegressor → Predictor
  158. def trainInternal[T <: VowpalWabbitBaseModel](dataset: Dataset[_], model: T): T

    Main training loop

    Main training loop

    dataset

    input data.

    returns

    binary VW model.

    Attributes
    protected
    Definition Classes
    VowpalWabbitBase
  159. def trainInternalDistributed(df: DataFrame, vwArgs: StringBuilder, numTasks: Int): Array[TrainingResult]

    Setup spanning tree and invoke training.

    Setup spanning tree and invoke training.

    df

    input data.

    vwArgs

    VW command line arguments.

    numTasks

    number of target tasks.

    Attributes
    protected
    Definition Classes
    VowpalWabbitBase
  160. def trainRow(schema: StructType, inputRows: Iterator[Row], ctx: TrainContext): Unit
    Attributes
    protected
    Definition Classes
    VowpalWabbitBase
  161. def transformSchema(schema: StructType): StructType
    Definition Classes
    Predictor → PipelineStage
  162. def transformSchema(schema: StructType, logging: Boolean): StructType
    Attributes
    protected
    Definition Classes
    PipelineStage
    Annotations
    @DeveloperApi()
  163. val uid: String
    Definition Classes
    VowpalWabbitRegressor → BasicLogging → Identifiable
  164. val useBarrierExecutionMode: BooleanParam
    Definition Classes
    VowpalWabbitBase
  165. def validateAndTransformSchema(schema: StructType, fitting: Boolean, featuresDataType: DataType): StructType
    Attributes
    protected
    Definition Classes
    PredictorParams
  166. val ver: String
    Definition Classes
    BasicLogging
  167. final def wait(): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  168. final def wait(arg0: Long, arg1: Int): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  169. final def wait(arg0: Long): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()
  170. val weightCol: Param[String]

    The name of the weight column

    The name of the weight column

    Definition Classes
    HasWeightCol
  171. def write: MLWriter
    Definition Classes
    ComplexParamsWritable → MLWritable

Inherited from BasicLogging

Inherited from ComplexParamsWritable

Inherited from MLWritable

Inherited from VowpalWabbitBase

Inherited from HasAdditionalFeatures

Inherited from HasWeightCol

Inherited from Wrappable

Inherited from RWrappable

Inherited from PythonWrappable

Inherited from BaseWrappable

Inherited from BaseRegressor[Row, VowpalWabbitRegressor, VowpalWabbitRegressionModel]

Inherited from Regressor[Row, VowpalWabbitRegressor, VowpalWabbitRegressionModel]

Inherited from Predictor[Row, VowpalWabbitRegressor, VowpalWabbitRegressionModel]

Inherited from PredictorParams

Inherited from HasPredictionCol

Inherited from HasFeaturesCol

Inherited from HasLabelCol

Inherited from Estimator[VowpalWabbitRegressionModel]

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