class VowpalWabbitContextualBandit extends Predictor[Row, VowpalWabbitContextualBandit, VowpalWabbitContextualBanditModel] with VowpalWabbitContextualBanditBase with HasParallelismInjected with ComplexParamsWritable with BasicLogging

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

Instance Constructors

  1. new VowpalWabbitContextualBandit()
  2. new VowpalWabbitContextualBandit(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
    VowpalWabbitContextualBanditVowpalWabbitBase
  6. val additionalFeatures: StringArrayParam
    Definition Classes
    HasAdditionalFeatures
  7. val additionalSharedFeatures: StringArrayParam
  8. val args: Param[String]
    Definition Classes
    VowpalWabbitBase
  9. final def asInstanceOf[T0]: T0
    Definition Classes
    Any
  10. def awaitFutures[T](futures: Array[Future[T]]): Seq[T]
    Attributes
    protected
    Definition Classes
    HasParallelismInjected
  11. val chosenActionCol: Param[String]
  12. lazy val classNameHelper: String
    Attributes
    protected
    Definition Classes
    BaseWrappable
  13. final def clear(param: Param[_]): VowpalWabbitContextualBandit.this.type
    Definition Classes
    Params
  14. def clone(): AnyRef
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()
  15. def companionModelClassName: String
    Attributes
    protected
    Definition Classes
    BaseWrappable
  16. def copy(extra: ParamMap): VowpalWabbitContextualBandit
    Definition Classes
    VowpalWabbitContextualBandit → Predictor → Estimator → PipelineStage → Params
  17. def copyValues[T <: Params](to: T, extra: ParamMap): T
    Attributes
    protected
    Definition Classes
    Params
  18. lazy val copyrightLines: String
    Attributes
    protected
    Definition Classes
    BaseWrappable
  19. def createLabelSetter(schema: StructType): (Row, VowpalWabbitExample) ⇒ Unit
    Attributes
    protected
    Definition Classes
    VowpalWabbitBase
  20. final def defaultCopy[T <: Params](extra: ParamMap): T
    Attributes
    protected
    Definition Classes
    Params
  21. val epsilon: DoubleParam
  22. final def eq(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  23. def equals(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  24. def explainParam(param: Param[_]): String
    Definition Classes
    Params
  25. def explainParams(): String
    Definition Classes
    Params
  26. def extractInstances(dataset: Dataset[_], validateInstance: (Instance) ⇒ Unit): RDD[Instance]
    Attributes
    protected
    Definition Classes
    PredictorParams
  27. def extractInstances(dataset: Dataset[_]): RDD[Instance]
    Attributes
    protected
    Definition Classes
    PredictorParams
  28. def extractLabeledPoints(dataset: Dataset[_]): RDD[LabeledPoint]
    Attributes
    protected
    Definition Classes
    Predictor
  29. final def extractParamMap(): ParamMap
    Definition Classes
    Params
  30. final def extractParamMap(extra: ParamMap): ParamMap
    Definition Classes
    Params
  31. final val featuresCol: Param[String]
    Definition Classes
    HasFeaturesCol
  32. def finalize(): Unit
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  33. def fit(dataset: Dataset[_], paramMaps: Seq[ParamMap]): Seq[VowpalWabbitContextualBanditModel]
    Definition Classes
    VowpalWabbitContextualBandit → Estimator
  34. def fit(dataset: Dataset[_]): VowpalWabbitContextualBanditModel
    Definition Classes
    Predictor → Estimator
  35. def fit(dataset: Dataset[_], paramMap: ParamMap): VowpalWabbitContextualBanditModel
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" )
  36. def fit(dataset: Dataset[_], firstParamPair: ParamPair[_], otherParamPairs: ParamPair[_]*): VowpalWabbitContextualBanditModel
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" ) @varargs()
  37. final def get[T](param: Param[T]): Option[T]
    Definition Classes
    Params
  38. def getAdditionalColumns: Seq[String]
    Attributes
    protected
    Definition Classes
    VowpalWabbitContextualBanditVowpalWabbitBase
  39. def getAdditionalFeatures: Array[String]
    Definition Classes
    HasAdditionalFeatures
  40. def getAdditionalSharedFeatures: Array[String]
  41. def getArgs: String
    Definition Classes
    VowpalWabbitBase
  42. def getAsFloat(schema: StructType, idx: Int): (Row) ⇒ Float
    Attributes
    protected
    Definition Classes
    VowpalWabbitBase
  43. def getChosenActionCol: String
  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 getEpsilon: Double
  47. def getExecutionContextProxy: ExecutionContext
    Definition Classes
    HasParallelismInjected
  48. final def getFeaturesCol: String
    Definition Classes
    HasFeaturesCol
  49. def getHashSeed: Int
    Definition Classes
    VowpalWabbitBase
  50. def getIgnoreNamespaces: String
    Definition Classes
    VowpalWabbitBase
  51. def getInitialModel: Array[Byte]
    Definition Classes
    VowpalWabbitBase
  52. def getInteractions: Array[String]
    Definition Classes
    VowpalWabbitBase
  53. def getL1: Double
    Definition Classes
    VowpalWabbitBase
  54. def getL2: Double
    Definition Classes
    VowpalWabbitBase
  55. final def getLabelCol: String
    Definition Classes
    HasLabelCol
  56. def getLearningRate: Double
    Definition Classes
    VowpalWabbitBase
  57. def getNumBits: Int
    Definition Classes
    VowpalWabbitBase
  58. def getNumPasses: Int
    Definition Classes
    VowpalWabbitBase
  59. final def getOrDefault[T](param: Param[T]): T
    Definition Classes
    Params
  60. def getParallelism: Int
    Definition Classes
    HasParallelism
  61. def getParam(paramName: String): Param[Any]
    Definition Classes
    Params
  62. def getPowerT: Double
    Definition Classes
    VowpalWabbitBase
  63. final def getPredictionCol: String
    Definition Classes
    HasPredictionCol
  64. def getProbabilityCol: String
  65. def getSharedCol: String
  66. def getUseBarrierExecutionMode: Boolean
    Definition Classes
    VowpalWabbitBase
  67. def getWeightCol: String

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

    Definition Classes
    HasWeightCol
  175. val sharedCol: Param[String]
  176. final def synchronized[T0](arg0: ⇒ T0): T0
    Definition Classes
    AnyRef
  177. def toString(): String
    Definition Classes
    Identifiable → AnyRef → Any
  178. def train(dataset: Dataset[_]): VowpalWabbitContextualBanditModel
    Attributes
    protected
    Definition Classes
    VowpalWabbitContextualBandit → Predictor
  179. 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
  180. 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
  181. def trainRow(schema: StructType, inputRows: Iterator[Row], ctx: TrainContext): Unit
    Attributes
    protected
    Definition Classes
    VowpalWabbitContextualBanditVowpalWabbitBase
  182. def transformSchema(schema: StructType): StructType
    Definition Classes
    VowpalWabbitContextualBandit → Predictor → PipelineStage
  183. def transformSchema(schema: StructType, logging: Boolean): StructType
    Attributes
    protected
    Definition Classes
    PipelineStage
    Annotations
    @DeveloperApi()
  184. val uid: String
    Definition Classes
    VowpalWabbitContextualBanditBasicLogging → Identifiable
  185. val useBarrierExecutionMode: BooleanParam
    Definition Classes
    VowpalWabbitBase
  186. def validateAndTransformSchema(schema: StructType, fitting: Boolean, featuresDataType: DataType): StructType
    Attributes
    protected
    Definition Classes
    PredictorParams
  187. val ver: String
    Definition Classes
    BasicLogging
  188. final def wait(): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  189. final def wait(arg0: Long, arg1: Int): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  190. final def wait(arg0: Long): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()
  191. val weightCol: Param[String]

    The name of the weight column

    The name of the weight column

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

Inherited from BasicLogging

Inherited from ComplexParamsWritable

Inherited from MLWritable

Inherited from HasParallelismInjected

Inherited from HasParallelism

Inherited from VowpalWabbitBase

Inherited from HasAdditionalFeatures

Inherited from HasWeightCol

Inherited from Wrappable

Inherited from RWrappable

Inherited from PythonWrappable

Inherited from BaseWrappable

Inherited from PredictorParams

Inherited from HasPredictionCol

Inherited from HasFeaturesCol

Inherited from HasLabelCol

Inherited from Estimator[VowpalWabbitContextualBanditModel]

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