Class/Object

org.apache.spark.ml.odkl

UnwrappedStage

Related Docs: object UnwrappedStage | package odkl

Permalink

class UnwrappedStage[M <: ModelWithSummary[M], T <: ModelTransformer[M, T]] extends Estimator[M] with SummarizableEstimator[M]

In case if we can avoid certain stages used during training while predicting we need to propagate some changes to the model (eg. unscale weights or remove intercept). Also useful for extending summary blocks (eg. during evaluation/cross-validation).

This class is used as a typical pipeline stage while training (fits and applies transformer, then calls the nested estimator), but it automatically eliminates itself from the resulting model by applying model transformer.

Linear Supertypes
SummarizableEstimator[M], Estimator[M], PipelineStage, Logging, Params, Serializable, Serializable, Identifiable, AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. UnwrappedStage
  2. SummarizableEstimator
  3. Estimator
  4. PipelineStage
  5. Logging
  6. Params
  7. Serializable
  8. Serializable
  9. Identifiable
  10. AnyRef
  11. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new UnwrappedStage(estimator: Estimator[M], transformer: T)

    Permalink

  2. new UnwrappedStage(estimator: Estimator[M], transformerTrainer: Estimator[T])

    Permalink

  3. new UnwrappedStage(estimator: Estimator[M], transformerTrainer: Estimator[T], uid: String)

    Permalink

    estimator

    Nested estimator to pass control

    transformerTrainer

    Estimator for fitting data AND model transformer.

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. def cacheTransformed: BooleanParam

    Permalink

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

    Permalink
    Definition Classes
    Params

  8. def clone(): AnyRef

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

  9. def copy(extra: ParamMap): SummarizableEstimator[M]

    Permalink
    Definition Classes
    UnwrappedStage → SummarizableEstimator → Estimator → PipelineStage → Params

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

    Permalink
    Attributes
    protected
    Definition Classes
    Params

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

    Permalink
    Attributes
    protected
    Definition Classes
    Params

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

    Permalink
    Definition Classes
    AnyRef

  13. def equals(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any

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

    Permalink
    Definition Classes
    Params

  15. def explainParams(): String

    Permalink
    Definition Classes
    Params

  16. final def extractParamMap(): ParamMap

    Permalink
    Definition Classes
    Params

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

    Permalink
    Definition Classes
    Params

  18. def finalize(): Unit

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

  19. def fit(dataset: Dataset[_]): M

    Permalink
    Definition Classes
    UnwrappedStage → Estimator

  20. def fit(dataset: Dataset[_], paramMaps: Array[ParamMap]): Seq[M]

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

  21. def fit(dataset: Dataset[_], paramMap: ParamMap): M

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

  22. def fit(dataset: Dataset[_], firstParamPair: ParamPair[_], otherParamPairs: ParamPair[_]*): M

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

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

    Permalink
    Definition Classes
    Params

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

    Permalink
    Definition Classes
    AnyRef → Any

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

    Permalink
    Definition Classes
    Params

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

    Permalink
    Definition Classes
    Params

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

    Permalink
    Definition Classes
    Params

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

    Permalink
    Definition Classes
    Params

  29. def hasParam(paramName: String): Boolean

    Permalink
    Definition Classes
    Params

  30. def hashCode(): Int

    Permalink
    Definition Classes
    AnyRef → Any

  31. def initializeLogIfNecessary(isInterpreter: Boolean): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

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

    Permalink
    Definition Classes
    Params

  33. final def isInstanceOf[T0]: Boolean

    Permalink
    Definition Classes
    Any

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

    Permalink
    Definition Classes
    Params

  35. def isTraceEnabled(): Boolean

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

  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. def materializeCached: BooleanParam

    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. final def set(paramPair: ParamPair[_]): UnwrappedStage.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    Params

  54. final def set(param: String, value: Any): UnwrappedStage.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    Params

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

    Permalink
    Definition Classes
    Params

  56. def setCacheTransformed(value: Boolean = true): UnwrappedStage.this.type

    Permalink

  57. final def setDefault(paramPairs: ParamPair[_]*): UnwrappedStage.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    Params

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

    Permalink
    Attributes
    protected
    Definition Classes
    Params

  59. def setMaterializeCached(value: Boolean = true): UnwrappedStage.this.type

    Permalink

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

    Permalink
    Definition Classes
    AnyRef

  61. def toString(): String

    Permalink
    Definition Classes
    Identifiable → AnyRef → Any

  62. def transformSchema(schema: StructType): StructType

    Permalink
    Definition Classes
    UnwrappedStage → PipelineStage
    Annotations
    @DeveloperApi()

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

    Permalink
    Attributes
    protected
    Definition Classes
    PipelineStage
    Annotations
    @DeveloperApi()

  64. val uid: String

    Permalink
    Definition Classes
    UnwrappedStage → Identifiable

  65. final def wait(): Unit

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

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

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

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

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

Inherited from SummarizableEstimator[M]

Inherited from Estimator[M]

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