Class

com.github.nearbydelta.deepspark.train

TrainerBuilder

Related Doc: package train

Permalink

class TrainerBuilder extends AnyRef

Builder for trainer.

Linear Supertypes
AnyRef, Any
Ordering
  1. Alphabetic
  2. By inheritance
Inherited
  1. TrainerBuilder
  2. AnyRef
  3. Any
  1. Hide All
  2. Show all
Visibility
  1. Public
  2. All

Instance Constructors

  1. new TrainerBuilder(param: TrainingParam = TrainingParam())

    Permalink

    param

    Training parameter

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 ==(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any
  4. final def asInstanceOf[T0]: T0

    Permalink
    Definition Classes
    Any
  5. def build[IN](network: Network[IN, DataVec], trainset: RDD[(IN, DataVec)], testset: RDD[(IN, DataVec)], objective: Objective, name: String)(implicit arg0: ClassTag[IN]): StaticTrainer[IN, DataVec]

    Permalink

    Train network

    Train network

    IN

    Type of input

    network

    Network

    trainset

    Train set

    testset

    Evaluation set

    objective

    Objective function

    name

    Name of trainer.

    returns

    trained network.

  6. def build[IN, OUT](network: Network[IN, OUT], trainset: RDD[(IN, OUT)], testset: RDD[(IN, OUT)], objective: Objective, expectConverter: (OUT) ⇒ DataVec, name: String)(implicit arg0: ClassTag[IN], arg1: ClassTag[OUT]): StaticTrainer[IN, OUT]

    Permalink

    Train network

    Train network

    IN

    Type of input

    OUT

    Type of real output

    network

    Network

    trainset

    Train set

    testset

    Evaluation set

    objective

    Objective function

    expectConverter

    Function to convert real output into vector

    name

    Name of trainer.

    returns

    trained network.

  7. def build[X, IN, OUT](network: Network[IN, OUT], trainset: RDD[X], testset: RDD[X], objective: Objective, expectConverter: (OUT) ⇒ DataVec, mapper: (X) ⇒ (IN, OUT), name: String = "temp")(implicit arg0: ClassTag[IN], arg1: ClassTag[OUT]): StaticTrainer[IN, OUT]

    Permalink

    Train network

    Train network

    X

    Type of given set

    IN

    Type of input

    OUT

    Type of real output

    network

    Network

    trainset

    Train set

    testset

    Evaluation set

    objective

    Objective function

    expectConverter

    Function to convert real output into vector

    mapper

    Function to convert item into (in, out) pair

    name

    Name of trainer.

    returns

    trained network.

  8. def buildFlexible[IN, OUT](network: Network[IN, OUT], trainset: RDD[(IN, OUT)], testset: RDD[(IN, OUT)], objective: Objective, expectConverter: (OUT) ⇒ DataVec, name: String)(implicit arg0: ClassTag[IN], arg1: ClassTag[OUT]): FlexibleTrainer[IN, OUT]

    Permalink

    Train network, real output vector is not fixed before evaluation.

    Train network, real output vector is not fixed before evaluation.

    IN

    Type of input

    OUT

    Type of real output

    network

    Network

    trainset

    Train set

    testset

    Evaluation set

    objective

    Objective function

    expectConverter

    Function to convert real output into vector

    name

    Name of trainer.

    returns

    trained network.

  9. def clone(): AnyRef

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

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

    Permalink
    Definition Classes
    AnyRef → Any
  12. def finalize(): Unit

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  13. final def getClass(): Class[_]

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

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

    Permalink
    Definition Classes
    Any
  16. final def ne(arg0: AnyRef): Boolean

    Permalink
    Definition Classes
    AnyRef
  17. final def notify(): Unit

    Permalink
    Definition Classes
    AnyRef
  18. final def notifyAll(): Unit

    Permalink
    Definition Classes
    AnyRef
  19. val param: TrainingParam

    Permalink

    Training parameter

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

    Permalink
    Definition Classes
    AnyRef
  21. def toString(): String

    Permalink
    Definition Classes
    AnyRef → Any
  22. final def wait(): Unit

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

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

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

Inherited from AnyRef

Inherited from Any

Ungrouped