VowpalWabbitClassifier

Instance Constructors

new VowpalWabbitClassifier()
new VowpalWabbitClassifier(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[_]): VowpalWabbitClassifier.this.type

Definition Classes
Params
def clone(): AnyRef

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

Definition Classes
VowpalWabbitClassifier → 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[_], numClasses: Int): RDD[LabeledPoint]

Attributes
protected
Definition Classes
Classifier
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[_]): VowpalWabbitClassificationModel

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

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

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

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 getNumClasses(dataset: Dataset[_], maxNumClasses: Int): Int

Attributes
protected
Definition Classes
Classifier
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
final def getProbabilityCol: String

Definition Classes
HasProbabilityCol
final def getRawPredictionCol: String

Definition Classes
HasRawPredictionCol
def getThresholds: Array[Double]

Definition Classes
HasThresholds
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
final val probabilityCol: Param[String]

Definition Classes
HasProbabilityCol
final val rawPredictionCol: Param[String]

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

Definition Classes
Predictor
def setProbabilityCol(value: String): VowpalWabbitClassifier

Definition Classes
ProbabilisticClassifier
def setRawPredictionCol(value: String): VowpalWabbitClassifier

Definition Classes
Classifier
def setThresholds(value: Array[Double]): VowpalWabbitClassifier

Definition Classes
ProbabilisticClassifier
def setWeightCol(value: String): VowpalWabbitClassifier.this.type

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

Definition Classes
AnyRef
final val thresholds: DoubleArrayParam

Definition Classes
HasThresholds
def toString(): String

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

Attributes
protected
Definition Classes
VowpalWabbitClassifier → 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
VowpalWabbitClassifier → Identifiable
def validateAndTransformSchema(schema: StructType, fitting: Boolean, featuresDataType: DataType): StructType

Attributes
protected
Definition Classes
ProbabilisticClassifierParams → ClassifierParams → 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 ProbabilisticClassifier[Row, VowpalWabbitClassifier, VowpalWabbitClassificationModel]

Inherited from ProbabilisticClassifierParams

Inherited from HasThresholds

Inherited from HasProbabilityCol

Inherited from Classifier[Row, VowpalWabbitClassifier, VowpalWabbitClassificationModel]

Inherited from ClassifierParams

Inherited from HasRawPredictionCol

Inherited from Predictor[Row, VowpalWabbitClassifier, VowpalWabbitClassificationModel]

Related Docs: object VowpalWabbitClassifier | package vw

class VowpalWabbitClassifier extends ProbabilisticClassifier[Row, VowpalWabbitClassifier, VowpalWabbitClassificationModel] with VowpalWabbitBase

Instance Constructors

new VowpalWabbitClassifier()

new VowpalWabbitClassifier(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[_]): VowpalWabbitClassifier.this.type

def clone(): AnyRef

def copy(extra: ParamMap): VowpalWabbitClassifier

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[_], numClasses: Int): RDD[LabeledPoint]

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[_]): VowpalWabbitClassificationModel

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

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

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

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 getNumClasses(dataset: Dataset[_], maxNumClasses: Int): 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

final def getProbabilityCol: String

final def getRawPredictionCol: String

def getThresholds: Array[Double]

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