org.apache.spark.ml.odkl

ForkedEstimator

abstract class ForkedEstimator[ModelIn <: ModelWithSummary[ModelIn], ForeKeyType, ModelOut <: ModelWithSummary[ModelOut]] extends Estimator[ModelOut] with SummarizableEstimator[ModelOut] with ForkedModelParams

Utility used to split training into forks (per type, per class, per fold).

ModelIn

Type of model produced by the nested estimator.

ModelOut

Type of the resulting model. Does not have to be the same as ModelIn.

Linear Supertypes
ForkedModelParams, SummarizableEstimator[ModelOut], Estimator[ModelOut], PipelineStage, Logging, Params, Serializable, Serializable, Identifiable, AnyRef, Any
Known Subclasses
Ordering
  1. Alphabetic
  2. By inheritance
Inherited
  1. ForkedEstimator
  2. ForkedModelParams
  3. SummarizableEstimator
  4. Estimator
  5. PipelineStage
  6. Logging
  7. Params
  8. Serializable
  9. Serializable
  10. Identifiable
  11. AnyRef
  12. Any
  1. Hide All
  2. Show all
Learn more about member selection
Visibility
  1. Public
  2. All

Instance Constructors

  1. new ForkedEstimator(nested: SummarizableEstimator[ModelIn], uid: String)

    nested

    Nested estimator to call for each fork.

Abstract Value Members

  1. abstract def copy(extra: ParamMap): SummarizableEstimator[ModelOut]

    Definition Classes
    SummarizableEstimator → Estimator → PipelineStage → Params
  2. abstract def createForks(dataset: DataFrame): Seq[(ForeKeyType, DataFrame)]

    Override this method and create forks to train from the data.

    Override this method and create forks to train from the data.

    Attributes
    protected
  3. abstract def mergeModels(sqlContext: SQLContext, models: Seq[(ForeKeyType, ModelIn)]): ModelOut

    Given models trained for each fork create a combined model.

    Given models trained for each fork create a combined model. This model is the result of the estimator.

    Attributes
    protected

Concrete Value Members

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

    Definition Classes
    AnyRef
  2. final def !=(arg0: Any): Boolean

    Definition Classes
    Any
  3. final def ##(): Int

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

    Attributes
    protected
    Definition Classes
    Params
  5. final def ==(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  6. final def ==(arg0: Any): Boolean

    Definition Classes
    Any
  7. final def asInstanceOf[T0]: T0

    Definition Classes
    Any
  8. final val cacheForks: Param[Boolean]

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

    Definition Classes
    Params
  10. def clone(): AnyRef

    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  11. def copyValues[T <: Params](to: T, extra: ParamMap): T

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

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

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

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

    Definition Classes
    Params
  16. def explainParams(): String

    Definition Classes
    Params
  17. final def extractParamMap(): ParamMap

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

    Definition Classes
    Params
  19. def finalize(): Unit

    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  20. def fit(dataset: DataFrame): ModelOut

    Definition Classes
    ForkedEstimator → Estimator
  21. def fit(dataset: DataFrame, paramMaps: Array[ParamMap]): Seq[ModelOut]

    Definition Classes
    Estimator
  22. def fit(dataset: DataFrame, paramMap: ParamMap): ModelOut

    Definition Classes
    Estimator
  23. def fit(dataset: DataFrame, firstParamPair: ParamPair[_], otherParamPairs: ParamPair[_]*): ModelOut

    Definition Classes
    Estimator
    Annotations
    @varargs()
  24. def fitFork(estimator: SummarizableEstimator[ModelIn], wholeData: DataFrame, partialData: (ForeKeyType, DataFrame)): (ForeKeyType, ModelIn)

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

    Definition Classes
    Params
  26. final def getClass(): Class[_]

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

    Definition Classes
    Params
  28. final def getOrDefault[T](param: Param[T]): T

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

    Definition Classes
    Params
  30. final def hasDefault[T](param: Param[T]): Boolean

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

    Definition Classes
    Params
  32. def hashCode(): Int

    Definition Classes
    AnyRef → Any
  33. final def isDefined(param: Param[_]): Boolean

    Definition Classes
    Params
  34. final def isInstanceOf[T0]: Boolean

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

    Definition Classes
    Params
  36. def isTraceEnabled(): Boolean

    Attributes
    protected
    Definition Classes
    Logging
  37. def log: Logger

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

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

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

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

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

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

    Attributes
    protected
    Definition Classes
    Logging
  44. def logName: String

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

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

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

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

    Attributes
    protected
    Definition Classes
    Logging
  49. def mayBePropagateKey(data: DataFrame, key: Any): DataFrame

    Attributes
    protected
    Definition Classes
    ForkedModelParams
  50. final def ne(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  51. val nested: SummarizableEstimator[ModelIn]

    Nested estimator to call for each fork.

  52. final def notify(): Unit

    Definition Classes
    AnyRef
  53. final def notifyAll(): Unit

    Definition Classes
    AnyRef
  54. lazy val params: Array[Param[_]]

    Definition Classes
    Params
  55. final val pathForTempModels: Param[String]

  56. final val propagatedKeyColumn: Param[String]

    Definition Classes
    ForkedModelParams
  57. final def set(paramPair: ParamPair[_]): ForkedEstimator.this.type

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

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

    Definition Classes
    Params
  60. def setCacheForks(value: Boolean): ForkedEstimator.this.type

  61. final def setDefault(paramPairs: ParamPair[_]*): ForkedEstimator.this.type

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

    Attributes
    protected
    Definition Classes
    Params
  63. def setPathForTempModels(value: String): ForkedEstimator.this.type

  64. def setPropagatedKeyColumn(value: String): ForkedEstimator.this.type

    Definition Classes
    ForkedModelParams
  65. def setTrainParallel(value: Boolean): ForkedEstimator.this.type

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

    Definition Classes
    AnyRef
  67. def toString(): String

    Definition Classes
    Identifiable → AnyRef → Any
  68. final val trainParallel: Param[Boolean]

  69. def transformSchema(schema: StructType): StructType

    Definition Classes
    ForkedEstimator → PipelineStage
    Annotations
    @DeveloperApi()
  70. def transformSchema(schema: StructType, logging: Boolean): StructType

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

    Definition Classes
    ForkedEstimator → Identifiable
  72. def validateParams(): Unit

    Definition Classes
    Params
  73. final def wait(): Unit

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

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

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

Inherited from ForkedModelParams

Inherited from SummarizableEstimator[ModelOut]

Inherited from Estimator[ModelOut]

Inherited from PipelineStage

Inherited from Logging

Inherited from Params

Inherited from Serializable

Inherited from Serializable

Inherited from Identifiable

Inherited from AnyRef

Inherited from Any

Ungrouped