VowpalWabbitRegressor

Instance Constructors

new VowpalWabbitRegressor()
new VowpalWabbitRegressor(uid: String)

Type Members

implicit class ParamStringBuilder extends AnyRef

Definition Classes
VowpalWabbitBase

Value Members

final def !=(arg0: Any): Boolean

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

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

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

Definition Classes
AnyRef → Any
val additionalFeatures: StringArrayParam

Definition Classes
HasAdditionalFeatures
val args: Param[String]

Definition Classes
VowpalWabbitBase
final def asInstanceOf[T0]: T0

Definition Classes
Any
final def clear(param: Param[_]): VowpalWabbitRegressor.this.type

Definition Classes
Params
def clone(): AnyRef

Attributes
protected[java.lang]
Definition Classes
AnyRef
Annotations
@throws( ... )
def copy(extra: ParamMap): VowpalWabbitRegressor

Definition Classes
VowpalWabbitRegressor → Predictor → Estimator → PipelineStage → Params
def copyValues[T <: Params](to: T, extra: ParamMap): T

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

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

Definition Classes
AnyRef
def equals(arg0: Any): Boolean

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

Definition Classes
Params
def explainParams(): String

Definition Classes
Params
def extractLabeledPoints(dataset: Dataset[_]): RDD[LabeledPoint]

Attributes
protected
Definition Classes
Predictor
final def extractParamMap(): ParamMap

Definition Classes
Params
final def extractParamMap(extra: ParamMap): ParamMap

Definition Classes
Params
final val featuresCol: Param[String]

Definition Classes
HasFeaturesCol
def finalize(): Unit

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

Definition Classes
Predictor → Estimator
def fit(dataset: Dataset[_], paramMaps: Array[ParamMap]): Seq[VowpalWabbitRegressorModel]

Definition Classes
Estimator
Annotations
@Since( "2.0.0" )
def fit(dataset: Dataset[_], paramMap: ParamMap): VowpalWabbitRegressorModel

Definition Classes
Estimator
Annotations
@Since( "2.0.0" )
def fit(dataset: Dataset[_], firstParamPair: ParamPair[_], otherParamPairs: ParamPair[_]*): VowpalWabbitRegressorModel

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

Definition Classes
Params
def getAdditionalFeatures: Array[String]

Definition Classes
HasAdditionalFeatures
def getArgs: String

Definition Classes
VowpalWabbitBase
final def getClass(): Class[_]

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

Definition Classes
Params
final def getFeaturesCol: String

Definition Classes
HasFeaturesCol
def getHashSeed: Int

Definition Classes
VowpalWabbitBase
def getIgnoreNamespaces: String

Definition Classes
VowpalWabbitBase
def getInitialModel: Array[Byte]

Definition Classes
VowpalWabbitBase
def getInteractions: Array[String]

Definition Classes
VowpalWabbitBase
def getL1: Double

Definition Classes
VowpalWabbitBase
def getL2: Double

Definition Classes
VowpalWabbitBase
final def getLabelCol: String

Definition Classes
HasLabelCol
def getLearningRate: Double

Definition Classes
VowpalWabbitBase
def getNumBits: Int

Definition Classes
VowpalWabbitBase
def getNumPasses: Int

Definition Classes
VowpalWabbitBase
final def getOrDefault[T](param: Param[T]): T

Definition Classes
Params
def getParam(paramName: String): Param[Any]

Definition Classes
Params
def getPowerT: Double

Definition Classes
VowpalWabbitBase
final def getPredictionCol: String

Definition Classes
HasPredictionCol
def getWeightCol: String

Definition Classes
HasWeightCol
final def hasDefault[T](param: Param[T]): Boolean

Definition Classes
Params
def hasParam(paramName: String): Boolean

Definition Classes
Params
def hashCode(): Int

Definition Classes
AnyRef → Any
val hashSeed: IntParam

Definition Classes
VowpalWabbitBase
val ignoreNamespaces: Param[String]

Definition Classes
VowpalWabbitBase
val initialModel: ByteArrayParam

Definition Classes
VowpalWabbitBase
def initializeLogIfNecessary(isInterpreter: Boolean, silent: Boolean): Boolean

Attributes
protected
Definition Classes
Logging
def initializeLogIfNecessary(isInterpreter: Boolean): Unit

Attributes
protected
Definition Classes
Logging
val interactions: StringArrayParam

Definition Classes
VowpalWabbitBase
final def isDefined(param: Param[_]): Boolean

Definition Classes
Params
final def isInstanceOf[T0]: Boolean

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

Definition Classes
Params
def isTraceEnabled(): Boolean

Attributes
protected
Definition Classes
Logging
val l1: DoubleParam

Definition Classes
VowpalWabbitBase
val l2: DoubleParam

Definition Classes
VowpalWabbitBase
final val labelCol: Param[String]

Definition Classes
HasLabelCol
val learningRate: DoubleParam

Definition Classes
VowpalWabbitBase
def log: Logger

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

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

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

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

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

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

Attributes
protected
Definition Classes
Logging
def logName: String

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

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

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

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

Attributes
protected
Definition Classes
Logging
final def ne(arg0: AnyRef): Boolean

Definition Classes
AnyRef
final def notify(): Unit

Definition Classes
AnyRef
final def notifyAll(): Unit

Definition Classes
AnyRef
val numBits: IntParam

Definition Classes
VowpalWabbitBase
val numPasses: IntParam

Definition Classes
VowpalWabbitBase
lazy val params: Array[Param[_]]

Definition Classes
Params
val powerT: DoubleParam

Definition Classes
VowpalWabbitBase
final val predictionCol: Param[String]

Definition Classes
HasPredictionCol
def save(path: String): Unit

Definition Classes
MLWritable
Annotations
@Since( "1.6.0" ) @throws( ... )
final def set(paramPair: ParamPair[_]): VowpalWabbitRegressor.this.type

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

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

Definition Classes
Params
def setAdditionalFeatures(value: Array[String]): VowpalWabbitRegressor.this.type

Definition Classes
HasAdditionalFeatures
def setArgs(value: String): VowpalWabbitRegressor.this.type

Definition Classes
VowpalWabbitBase
final def setDefault(paramPairs: ParamPair[_]*): VowpalWabbitRegressor.this.type

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

Attributes
protected
Definition Classes
Params
def setFeaturesCol(value: String): VowpalWabbitRegressor

Definition Classes
Predictor
def setHashSeed(value: Int): VowpalWabbitRegressor.this.type

Definition Classes
VowpalWabbitBase
def setIgnoreNamespaces(value: String): VowpalWabbitRegressor.this.type

Definition Classes
VowpalWabbitBase
def setInitialModel(value: Array[Byte]): VowpalWabbitRegressor.this.type

Definition Classes
VowpalWabbitBase
def setInteractions(value: Array[String]): VowpalWabbitRegressor.this.type

Definition Classes
VowpalWabbitBase
def setL1(value: Double): VowpalWabbitRegressor.this.type

Definition Classes
VowpalWabbitBase
def setL2(value: Double): VowpalWabbitRegressor.this.type

Definition Classes
VowpalWabbitBase
def setLabelCol(value: String): VowpalWabbitRegressor

Definition Classes
Predictor
def setLearningRate(value: Double): VowpalWabbitRegressor.this.type

Definition Classes
VowpalWabbitBase
def setNumBits(value: Int): VowpalWabbitRegressor.this.type

Definition Classes
VowpalWabbitBase
def setNumPasses(value: Int): VowpalWabbitRegressor.this.type

Definition Classes
VowpalWabbitBase
def setPowerT(value: Double): VowpalWabbitRegressor.this.type

Definition Classes
VowpalWabbitBase
def setPredictionCol(value: String): VowpalWabbitRegressor

Definition Classes
Predictor
def setWeightCol(value: String): VowpalWabbitRegressor.this.type

Definition Classes
HasWeightCol
final def synchronized[T0](arg0: ⇒ T0): T0

Definition Classes
AnyRef
def toString(): String

Definition Classes
Identifiable → AnyRef → Any
def train(dataset: Dataset[_]): VowpalWabbitRegressorModel

Definition Classes
VowpalWabbitRegressor → Predictor
def trainInternal(dataset: Dataset[_]): Array[Byte]

Main training loop
Main training loop
dataset
input data.
returns
binary VW model.

Attributes
protected
Definition Classes
VowpalWabbitBase
def trainInternalDistributed(df: DataFrame, vwArgs: StringBuilder, numWorkers: Int): Option[Array[Byte]]

Setup spanning tree and invoke training.
Setup spanning tree and invoke training.
df
input data.
vwArgs
VW command line arguments.
numWorkers
number of target workers.

Attributes
protected
Definition Classes
VowpalWabbitBase
def transformSchema(schema: StructType): StructType

Definition Classes
Predictor → PipelineStage
def transformSchema(schema: StructType, logging: Boolean): StructType

Attributes
protected
Definition Classes
PipelineStage
Annotations
@DeveloperApi()
val uid: String

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

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

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

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

Definition Classes
AnyRef
Annotations
@throws( ... )
val weightCol: Param[String]

The name of the weight column
The name of the weight column

Definition Classes
HasWeightCol
def write: MLWriter

Definition Classes
DefaultParamsWritable → MLWritable

Inherited from VowpalWabbitBase

Inherited from HasAdditionalFeatures

Inherited from HasWeightCol

Inherited from DefaultParamsWritable

Inherited from MLWritable

Inherited from Wrappable

Inherited from BaseRegressor[Row, VowpalWabbitRegressor, VowpalWabbitRegressorModel]

Inherited from Regressor[Row, VowpalWabbitRegressor, VowpalWabbitRegressorModel]

Inherited from Predictor[Row, VowpalWabbitRegressor, VowpalWabbitRegressorModel]

Related Docs: object VowpalWabbitRegressor | package vw

class VowpalWabbitRegressor extends BaseRegressor[Row, VowpalWabbitRegressor, VowpalWabbitRegressorModel] with VowpalWabbitBase

Instance Constructors

new VowpalWabbitRegressor()

new VowpalWabbitRegressor(uid: String)

Type Members

implicit class ParamStringBuilder extends AnyRef

Value Members

final def !=(arg0: Any): Boolean

final def ##(): Int

final def $[T](param: Param[T]): T

final def ==(arg0: Any): Boolean

val additionalFeatures: StringArrayParam

val args: Param[String]

final def asInstanceOf[T0]: T0

final def clear(param: Param[_]): VowpalWabbitRegressor.this.type

def clone(): AnyRef

def copy(extra: ParamMap): VowpalWabbitRegressor

def copyValues[T <: Params](to: T, extra: ParamMap): T

final def defaultCopy[T <: Params](extra: ParamMap): T

final def eq(arg0: AnyRef): Boolean

def equals(arg0: Any): Boolean

def explainParam(param: Param[_]): String

def explainParams(): String

def extractLabeledPoints(dataset: Dataset[_]): RDD[LabeledPoint]

final def extractParamMap(): ParamMap

final def extractParamMap(extra: ParamMap): ParamMap

final val featuresCol: Param[String]

def finalize(): Unit

def fit(dataset: Dataset[_]): VowpalWabbitRegressorModel

def fit(dataset: Dataset[_], paramMaps: Array[ParamMap]): Seq[VowpalWabbitRegressorModel]

def fit(dataset: Dataset[_], paramMap: ParamMap): VowpalWabbitRegressorModel

def fit(dataset: Dataset[_], firstParamPair: ParamPair[_], otherParamPairs: ParamPair[_]*): VowpalWabbitRegressorModel

final def get[T](param: Param[T]): Option[T]

def getAdditionalFeatures: Array[String]

def getArgs: String

final def getClass(): Class[_]

final def getDefault[T](param: Param[T]): Option[T]

final def getFeaturesCol: String

def getHashSeed: Int

def getIgnoreNamespaces: String

def getInitialModel: Array[Byte]

def getInteractions: Array[String]

def getL1: Double

def getL2: Double

final def getLabelCol: String

def getLearningRate: Double

def getNumBits: Int

def getNumPasses: Int

final def getOrDefault[T](param: Param[T]): T

def getParam(paramName: String): Param[Any]

def getPowerT: Double

final def getPredictionCol: String

def getWeightCol: String

final def hasDefault[T](param: Param[T]): Boolean

def hasParam(paramName: String): Boolean

def hashCode(): Int

val hashSeed: IntParam

val ignoreNamespaces: Param[String]

val initialModel: ByteArrayParam

def initializeLogIfNecessary(isInterpreter: Boolean, silent: Boolean): Boolean

def initializeLogIfNecessary(isInterpreter: Boolean): Unit

val interactions: StringArrayParam

final def isDefined(param: Param[_]): Boolean

final def isInstanceOf[T0]: Boolean

final def isSet(param: Param[_]): Boolean

def isTraceEnabled(): Boolean

val l1: DoubleParam

val l2: DoubleParam

final val labelCol: Param[String]

val learningRate: DoubleParam

def log: Logger

def logDebug(msg: ⇒ String, throwable: Throwable): Unit

def logDebug(msg: ⇒ String): Unit

def logError(msg: ⇒ String, throwable: Throwable): Unit

def logError(msg: ⇒ String): Unit

def logInfo(msg: ⇒ String, throwable: Throwable): Unit

def logInfo(msg: ⇒ String): Unit

def logName: String