kr.ac.kaist.ir.deep.train

MultiThreadTrainStyle

class MultiThreadTrainStyle[IN, OUT] extends TrainStyle[IN, OUT]

Trainer : Stochastic-Style, Multi-Threaded using Spark.

Note

This is not a implementation using DistBelief Paper. This is between DistBeliefTrainStyle(DBTS) and SingleThreadTrainStyle(STTS). The major difference is whether "updating" is asynchronous(DBTS) or not(MTTS).

Linear Supertypes
TrainStyle[IN, OUT], Serializable, Serializable, AnyRef, Any
Known Subclasses
Ordering
  1. Alphabetic
  2. By inheritance
Inherited
  1. MultiThreadTrainStyle
  2. TrainStyle
  3. Serializable
  4. Serializable
  5. AnyRef
  6. Any
  1. Hide All
  2. Show all
Learn more about member selection
Visibility
  1. Public
  2. All

Instance Constructors

  1. new MultiThreadTrainStyle(net: Network, algorithm: WeightUpdater, sc: SparkContext, make: ManipulationType[IN, OUT] = new VectorType(), param: DistBeliefCriteria = DistBeliefCriteria())(implicit arg0: ClassTag[IN], arg1: ClassTag[OUT])

    net

    Network to be trained

    algorithm

    Weight update algorithm to be applied

    make

    Input Operation that supervises how to manipulate input as matrices. This also controls how to compute actual network. (default: VectorType)

    param

    Training criteria (default: SimpleTrainingCriteria)

Type Members

  1. type Pair = (IN, OUT)

    Training Pair Type

    Training Pair Type

    Definition Classes
    TrainStyle
  2. type Sampler = (Int) ⇒ Seq[OUT]

    Sampler Type

    Sampler Type

    Definition Classes
    TrainStyle
  3. implicit class WeightOp extends Serializable

    Implicit weight operation

    Implicit weight operation

    Definition Classes
    TrainStyle

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. val accCount: Accumulator[Int]

    Accumulator variable for counter

    Accumulator variable for counter

    Attributes
    protected
  5. val accNet: Accumulator[IndexedSeq[ScalarMatrix]]

    Accumulator variable for networks

    Accumulator variable for networks

    Attributes
    protected
  6. val algorithm: WeightUpdater

    Weight update algorithm to be applied

    Weight update algorithm to be applied

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

    Definition Classes
    Any
  8. def batch(): Unit

    Do mini-batch

    Do mini-batch

    Definition Classes
    MultiThreadTrainStyleTrainStyle
  9. var bcNet: Broadcast[Network]

    Spark distributed networks

    Spark distributed networks

    Attributes
    protected
  10. def clone(): AnyRef

    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  11. final def eq(arg0: AnyRef): Boolean

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

    Definition Classes
    AnyRef → Any
  13. def fetch(iter: Int): Unit

    Fetch weights

    Fetch weights

    iter

    current iteration

    Definition Classes
    MultiThreadTrainStyleTrainStyle
  14. def finalize(): Unit

    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  15. def foreachTestSet(n: Int)(fn: ((IN, OUT)) ⇒ Unit): Unit

    Iterate over given number of test instances

    Iterate over given number of test instances

    n

    number of random sampled instances

    fn

    iteratee function

    Definition Classes
    MultiThreadTrainStyleTrainStyle
  16. final def getClass(): Class[_]

    Definition Classes
    AnyRef → Any
  17. def hashCode(): Int

    Definition Classes
    AnyRef → Any
  18. final def isInstanceOf[T0]: Boolean

    Definition Classes
    Any
  19. def isUpdateFinished: Future[_]

    Indicates whether the asynchronous update is finished or not.

    Indicates whether the asynchronous update is finished or not.

    returns

    future object of update

    Definition Classes
    TrainStyle
  20. val logger: Logger

    Logger

    Logger

    Attributes
    protected
    Definition Classes
    TrainStyle
  21. val make: ManipulationType[IN, OUT]

    Input Operation that supervises how to manipulate input as matrices.

    Input Operation that supervises how to manipulate input as matrices. This also controls how to compute actual network. (default: VectorType)

    Definition Classes
    MultiThreadTrainStyleTrainStyle
  22. final def ne(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  23. var negFraction: Float

    Fraction of negative samples

    Fraction of negative samples

    Attributes
    protected
  24. var negOutUniverse: RDD[(Long, OUT)]

    Negative Sampler

    Negative Sampler

    Attributes
    protected
  25. var negPartitioner: RandomEqualPartitioner

    Partitioner for negative samples

    Partitioner for negative samples

    Attributes
    protected
  26. val net: Network

    Network to be trained

    Network to be trained

    Definition Classes
    MultiThreadTrainStyleTrainStyle
  27. final def notify(): Unit

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

    Definition Classes
    AnyRef
  29. val param: DistBeliefCriteria

    Training criteria (default: SimpleTrainingCriteria)

    Training criteria (default: SimpleTrainingCriteria)

    Definition Classes
    MultiThreadTrainStyleTrainStyle
  30. val sc: SparkContext

  31. def setNegativeTrainingReference(set: RDD[OUT]): Unit

    Set negative sampling method.

    Set negative sampling method.

    set

    all training outputs that will be used for negative training

    Definition Classes
    MultiThreadTrainStyleTrainStyle
  32. def setNegativeTrainingReference(set: Seq[OUT]): Unit

    Set negative sampling method.

    Set negative sampling method.

    set

    all training outputs that will be used for negative training

    Definition Classes
    MultiThreadTrainStyleTrainStyle
  33. def setPositiveTrainingReference(set: RDD[(IN, OUT)]): Unit

    Set training instances

    Set training instances

    set

    RDD of training set

    Definition Classes
    MultiThreadTrainStyleTrainStyle
  34. def setPositiveTrainingReference(set: Seq[(IN, OUT)]): Unit

    Set training instances

    Set training instances

    set

    Sequence of training set

    Definition Classes
    MultiThreadTrainStyleTrainStyle
  35. def setTestReference(set: RDD[(IN, OUT)]): Unit

    Set testing instances

    Set testing instances

    set

    RDD of testing set

    Definition Classes
    MultiThreadTrainStyleTrainStyle
  36. def setTestReference(set: Seq[(IN, OUT)]): Unit

    Set testing instances

    Set testing instances

    set

    Sequence of testing set

    Definition Classes
    MultiThreadTrainStyleTrainStyle
  37. def stopUntilBatchFinished(): Unit

    Non-blocking pending, until all assigned batches are finished

    Non-blocking pending, until all assigned batches are finished

    Definition Classes
    TrainStyle
  38. final def synchronized[T0](arg0: ⇒ T0): T0

    Definition Classes
    AnyRef
  39. var testSet: RDD[Pair]

    Test Set

    Test Set

    Attributes
    protected
  40. var testSize: Float

    Size of test set

    Size of test set

    Attributes
    protected
  41. def toString(): String

    Definition Classes
    AnyRef → Any
  42. var trainingFraction: Float

    Fraction of mini-batch

    Fraction of mini-batch

    Attributes
    protected
  43. var trainingSet: RDD[Pair]

    Training set

    Training set

    Attributes
    protected
  44. def unpersist(): Unit

    Unpersist all

  45. def update(iter: Int): Unit

    Send update of weights

    Send update of weights

    iter

    current iteration

    Definition Classes
    MultiThreadTrainStyleTrainStyle
  46. def validationError(): Float

    Calculate validation error

    Calculate validation error

    returns

    validation error

    Definition Classes
    MultiThreadTrainStyleTrainStyle
  47. final def wait(): Unit

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

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

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

Inherited from TrainStyle[IN, OUT]

Inherited from Serializable

Inherited from Serializable

Inherited from AnyRef

Inherited from Any

Ungrouped