kr.ac.kaist.ir.deep.train

DistBeliefTrainStyle

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

Train Style : Semi-DistBelief Style, Spark-based.

Note

Unlike with DistBelief, this trainer do updates and fetch by master not the workers.

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

Instance Constructors

  1. new DistBeliefTrainStyle(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

    sc

    A spark context that network will be distributed

    make

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

    param

    DistBelief-style Training criteria (default: DistBeliefCriteria)

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
    Definition Classes
    MultiThreadTrainStyle
  5. val accNet: Accumulator[IndexedSeq[ScalarMatrix]]

    Accumulator variable for networks

    Accumulator variable for networks

    Attributes
    protected
    Definition Classes
    MultiThreadTrainStyle
  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
    DistBeliefTrainStyleMultiThreadTrainStyleTrainStyle
  9. var batchFlag: ArrayBuffer[Future[Unit]]

    Flag for batch : Is Batch remaining?

    Flag for batch : Is Batch remaining?

    Attributes
    protected
  10. var bcNet: Broadcast[Network]

    Spark distributed networks

    Spark distributed networks

    Attributes
    protected
    Definition Classes
    MultiThreadTrainStyle
  11. def clone(): AnyRef

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

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

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

    Fetch weights

    Fetch weights

    iter

    current iteration

    Definition Classes
    DistBeliefTrainStyleMultiThreadTrainStyleTrainStyle
  15. var fetchFlag: Future[Unit]

    Flag for fetch : Is fetching?

    Flag for fetch : Is fetching?

    Attributes
    protected
  16. def finalize(): Unit

    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  17. 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
  18. final def getClass(): Class[_]

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

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

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

    Indicates whether the asynchrononus update is finished or not.

    Indicates whether the asynchrononus update is finished or not.

    returns

    future object of update

    Definition Classes
    DistBeliefTrainStyleTrainStyle
  22. val logger: Logger

    Logger

    Logger

    Attributes
    protected
    Definition Classes
    TrainStyle
  23. 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
  24. final def ne(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  25. var negFraction: Float

    Fraction of negative samples

    Fraction of negative samples

    Attributes
    protected
    Definition Classes
    MultiThreadTrainStyle
  26. var negOutUniverse: RDD[(Long, OUT)]

    Negative Sampler

    Negative Sampler

    Attributes
    protected
    Definition Classes
    MultiThreadTrainStyle
  27. var negPartitioner: RandomEqualPartitioner

    Partitioner for negative samples

    Partitioner for negative samples

    Attributes
    protected
    Definition Classes
    MultiThreadTrainStyle
  28. val net: Network

    Network to be trained

    Network to be trained

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

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

    Definition Classes
    AnyRef
  31. val param: DistBeliefCriteria

    Training criteria (default: SimpleTrainingCriteria)

    Training criteria (default: SimpleTrainingCriteria)

    Definition Classes
    MultiThreadTrainStyleTrainStyle
  32. val sc: SparkContext

    Definition Classes
    MultiThreadTrainStyle
  33. 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
  34. 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
  35. def setPositiveTrainingReference(set: RDD[(IN, OUT)]): Unit

    Set training instances

    Set training instances

    set

    RDD of training set

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

    Set training instances

    Set training instances

    set

    Sequence of training set

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

    Set testing instances

    Set testing instances

    set

    RDD of testing set

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

    Set testing instances

    Set testing instances

    set

    Sequence of testing set

    Definition Classes
    MultiThreadTrainStyleTrainStyle
  39. def stopUntilBatchFinished(): Unit

    Non-blocking pending, until all assigned batches are finished

    Non-blocking pending, until all assigned batches are finished

    Definition Classes
    DistBeliefTrainStyleTrainStyle
  40. final def synchronized[T0](arg0: ⇒ T0): T0

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

    Test Set

    Test Set

    Attributes
    protected
    Definition Classes
    MultiThreadTrainStyle
  42. var testSize: Float

    Size of test set

    Size of test set

    Attributes
    protected
    Definition Classes
    MultiThreadTrainStyle
  43. def toString(): String

    Definition Classes
    AnyRef → Any
  44. var trainingFraction: Float

    Fraction of mini-batch

    Fraction of mini-batch

    Attributes
    protected
    Definition Classes
    MultiThreadTrainStyle
  45. var trainingSet: RDD[Pair]

    Training set

    Training set

    Attributes
    protected
    Definition Classes
    MultiThreadTrainStyle
  46. def unpersist(): Unit

    Unpersist all

    Unpersist all

    Definition Classes
    MultiThreadTrainStyle
  47. def update(iter: Int): Unit

    Send update of weights

    Send update of weights

    iter

    current iteration

    Definition Classes
    DistBeliefTrainStyleMultiThreadTrainStyleTrainStyle
  48. var updateFlag: Future[Unit]

    Flag for update : Is updating?

    Flag for update : Is updating?

    Attributes
    protected
  49. var validationEpoch: Int

    size of training set

    size of training set

    Definition Classes
    TrainStyle
  50. def validationError(): Float

    Calculate validation error

    Calculate validation error

    returns

    validation error

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

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

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

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

Inherited from MultiThreadTrainStyle[IN, OUT]

Inherited from TrainStyle[IN, OUT]

Inherited from Serializable

Inherited from Serializable

Inherited from AnyRef

Inherited from Any

Ungrouped