case class TrainingContext(batchIndex: Int, sharedStateSingleton: SharedSingleton[SharedState], schema: StructType, numCols: Int, numInitScoreClasses: Int, trainingParams: BaseTrainParams, networkParams: NetworkParams, columnParams: ColumnParams, datasetParams: String, featureNames: Option[Array[String]], numTasksPerExecutor: Int, validationData: Option[Broadcast[Array[Row]]], serializedReferenceDataset: Option[Array[Byte]], partitionCounts: Option[Array[Long]]) extends Serializable with Product

Object to encapsulate all information about a training session that does not change during execution and can be created on the driver. There is also a reference to the shared state in an executor, which can change over time.

Linear Supertypes
Product, Equals, Serializable, Serializable, AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. TrainingContext
  2. Product
  3. Equals
  4. Serializable
  5. Serializable
  6. AnyRef
  7. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new TrainingContext(batchIndex: Int, sharedStateSingleton: SharedSingleton[SharedState], schema: StructType, numCols: Int, numInitScoreClasses: Int, trainingParams: BaseTrainParams, networkParams: NetworkParams, columnParams: ColumnParams, datasetParams: String, featureNames: Option[Array[String]], numTasksPerExecutor: Int, validationData: Option[Broadcast[Array[Row]]], serializedReferenceDataset: Option[Array[Byte]], partitionCounts: Option[Array[Long]])

Value Members

  1. final def !=(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  2. final def ##(): Int
    Definition Classes
    AnyRef → Any
  3. final def ==(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  4. final def asInstanceOf[T0]: T0
    Definition Classes
    Any
  5. val batchIndex: Int
  6. def clone(): AnyRef
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()
  7. val columnParams: ColumnParams
  8. val datasetParams: String
  9. val earlyStoppingRound: Int
  10. final def eq(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  11. val featureNames: Option[Array[String]]
  12. def finalize(): Unit
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  13. final def getClass(): Class[_]
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  14. val hasGroups: Boolean
  15. val hasGroupsAsInt: Int
  16. val hasInitialScores: Boolean
  17. val hasInitialScoresAsInt: Int
  18. val hasValidationData: Boolean
  19. val hasWeights: Boolean
  20. val hasWeightsAsInt: Int
  21. val improvementTolerance: Double
  22. def incrementArrayProcessedSignal(log: Logger): Int
  23. def incrementDataPrepDoneSignal(log: Logger): Unit
  24. val isBulk: Boolean
  25. val isClassification: Boolean
  26. final def isInstanceOf[T0]: Boolean
    Definition Classes
    Any
  27. val isProvideTrainingMetric: Boolean
  28. val isStreaming: Boolean
  29. val microBatchSize: Int
  30. final def ne(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  31. val networkParams: NetworkParams
  32. final def notify(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  33. final def notifyAll(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  34. val numCols: Int
  35. val numInitScoreClasses: Int
  36. val numTasksPerExecutor: Int
  37. val partitionCounts: Option[Array[Long]]
  38. val schema: StructType
  39. val serializedReferenceDataset: Option[Array[Byte]]
  40. def sharedState(): SharedState
  41. val sharedStateSingleton: SharedSingleton[SharedState]
  42. def shouldCreateValidationDataset(): Boolean

    Determines if the current task should calculate the validation Dataset.

    Determines if the current task should calculate the validation Dataset. Only 1 task per executor needs to do it, and first one to call this gets the assignment.

    returns

    True if the current task should create, false otherwise.

  43. final def synchronized[T0](arg0: ⇒ T0): T0
    Definition Classes
    AnyRef
  44. implicit def toInt(b: Boolean): Int
  45. val trainingParams: BaseTrainParams
  46. val useSingleDatasetMode: Boolean
  47. val validationData: Option[Broadcast[Array[Row]]]
  48. final def wait(): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  49. final def wait(arg0: Long, arg1: Int): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  50. final def wait(arg0: Long): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()

Inherited from Product

Inherited from Equals

Inherited from Serializable

Inherited from Serializable

Inherited from AnyRef

Inherited from Any

Ungrouped