Class/Object

com.microsoft.ml.spark.stages

StratifiedRepartition

Related Docs: object StratifiedRepartition | package stages

Permalink

class StratifiedRepartition extends Transformer with Wrappable with DefaultParamsWritable with HasLabelCol with HasSeed

StratifiedRepartition repartitions the DataFrame such that each label is selected in each partition. This may be necessary in some cases such as in LightGBM multiclass classification, where it is necessary for at least one instance of each label to be present on each partition.

Linear Supertypes
HasSeed, HasLabelCol, DefaultParamsWritable, MLWritable, Wrappable, Transformer, PipelineStage, Logging, Params, Serializable, Serializable, Identifiable, AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. StratifiedRepartition
  2. HasSeed
  3. HasLabelCol
  4. DefaultParamsWritable
  5. MLWritable
  6. Wrappable
  7. Transformer
  8. PipelineStage
  9. Logging
  10. Params
  11. Serializable
  12. Serializable
  13. Identifiable
  14. AnyRef
  15. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new StratifiedRepartition()

    Permalink

  2. new StratifiedRepartition(uid: String)

    Permalink

Value Members

  1. final def !=(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any

  2. final def ##(): Int

    Permalink
    Definition Classes
    AnyRef → Any

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

    Permalink
    Attributes
    protected
    Definition Classes
    Params

  4. final def ==(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any

  5. final def asInstanceOf[T0]: T0

    Permalink
    Definition Classes
    Any

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

    Permalink
    Definition Classes
    Params

  7. def clone(): AnyRef

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

  8. def copy(extra: ParamMap): DropColumns

    Permalink
    Definition Classes
    StratifiedRepartition → Transformer → PipelineStage → Params

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

    Permalink
    Attributes
    protected
    Definition Classes
    Params

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

    Permalink
    Attributes
    protected
    Definition Classes
    Params

  11. final def eq(arg0: AnyRef): Boolean

    Permalink
    Definition Classes
    AnyRef

  12. def equals(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any

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

    Permalink
    Definition Classes
    Params

  14. def explainParams(): String

    Permalink
    Definition Classes
    Params

  15. final def extractParamMap(): ParamMap

    Permalink
    Definition Classes
    Params

  16. final def extractParamMap(extra: ParamMap): ParamMap

    Permalink
    Definition Classes
    Params

  17. def finalize(): Unit

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )

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

    Permalink
    Definition Classes
    Params

  19. final def getClass(): Class[_]

    Permalink
    Definition Classes
    AnyRef → Any

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

    Permalink
    Definition Classes
    Params

  21. def getLabelCol: String

    Permalink

    Definition Classes
    HasLabelCol

  22. def getMode: String

    Permalink

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

    Permalink
    Definition Classes
    Params

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

    Permalink
    Definition Classes
    Params

  25. final def getSeed: Long

    Permalink
    Definition Classes
    HasSeed

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

    Permalink
    Definition Classes
    Params

  27. def hasParam(paramName: String): Boolean

    Permalink
    Definition Classes
    Params

  28. def hashCode(): Int

    Permalink
    Definition Classes
    AnyRef → Any

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

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

  30. def initializeLogIfNecessary(isInterpreter: Boolean): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

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

    Permalink
    Definition Classes
    Params

  32. final def isInstanceOf[T0]: Boolean

    Permalink
    Definition Classes
    Any

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

    Permalink
    Definition Classes
    Params

  34. def isTraceEnabled(): Boolean

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

  35. val labelCol: Param[String]

    Permalink

    The name of the label column

    The name of the label column

    Definition Classes
    HasLabelCol

  36. def log: Logger

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

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

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

  38. def logDebug(msg: ⇒ String): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

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

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

  40. def logError(msg: ⇒ String): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

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

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

  42. def logInfo(msg: ⇒ String): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

  43. def logName: String

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

  44. def logTrace(msg: ⇒ String, throwable: Throwable): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

  45. def logTrace(msg: ⇒ String): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

  46. def logWarning(msg: ⇒ String, throwable: Throwable): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

  47. def logWarning(msg: ⇒ String): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

  48. val mode: Param[String]

    Permalink

  49. final def ne(arg0: AnyRef): Boolean

    Permalink
    Definition Classes
    AnyRef

  50. final def notify(): Unit

    Permalink
    Definition Classes
    AnyRef

  51. final def notifyAll(): Unit

    Permalink
    Definition Classes
    AnyRef

  52. lazy val params: Array[Param[_]]

    Permalink
    Definition Classes
    Params

  53. def save(path: String): Unit

    Permalink
    Definition Classes
    MLWritable
    Annotations
    @Since( "1.6.0" ) @throws( ... )

  54. final val seed: LongParam

    Permalink
    Definition Classes
    HasSeed

  55. final def set(paramPair: ParamPair[_]): StratifiedRepartition.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    Params

  56. final def set(param: String, value: Any): StratifiedRepartition.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    Params

  57. final def set[T](param: Param[T], value: T): StratifiedRepartition.this.type

    Permalink
    Definition Classes
    Params

  58. final def setDefault(paramPairs: ParamPair[_]*): StratifiedRepartition.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    Params

  59. final def setDefault[T](param: Param[T], value: T): StratifiedRepartition.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    Params

  60. def setLabelCol(value: String): StratifiedRepartition.this.type

    Permalink

    Definition Classes
    HasLabelCol

  61. def setMode(value: String): StratifiedRepartition.this.type

    Permalink

  62. final def synchronized[T0](arg0: ⇒ T0): T0

    Permalink
    Definition Classes
    AnyRef

  63. def toString(): String

    Permalink
    Definition Classes
    Identifiable → AnyRef → Any

  64. def transform(dataset: Dataset[_]): DataFrame

    Permalink

    dataset

    - The input dataset, to be transformed

    returns

    The DataFrame that results from stratified repartitioning

    Definition Classes
    StratifiedRepartition → Transformer

  65. def transform(dataset: Dataset[_], paramMap: ParamMap): DataFrame

    Permalink
    Definition Classes
    Transformer
    Annotations
    @Since( "2.0.0" )

  66. def transform(dataset: Dataset[_], firstParamPair: ParamPair[_], otherParamPairs: ParamPair[_]*): DataFrame

    Permalink
    Definition Classes
    Transformer
    Annotations
    @Since( "2.0.0" ) @varargs()

  67. def transformSchema(schema: StructType): StructType

    Permalink
    Definition Classes
    StratifiedRepartition → PipelineStage

  68. def transformSchema(schema: StructType, logging: Boolean): StructType

    Permalink
    Attributes
    protected
    Definition Classes
    PipelineStage
    Annotations
    @DeveloperApi()

  69. val uid: String

    Permalink
    Definition Classes
    StratifiedRepartition → Identifiable

  70. final def wait(): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

  71. final def wait(arg0: Long, arg1: Int): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

  72. final def wait(arg0: Long): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

  73. def write: MLWriter

    Permalink
    Definition Classes
    DefaultParamsWritable → MLWritable

Inherited from HasSeed

Inherited from HasLabelCol

Inherited from DefaultParamsWritable

Inherited from MLWritable

Inherited from Wrappable

Inherited from Transformer

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