Class

org.apache.mxnet.FeedForward

Builder

Related Doc: package FeedForward

Permalink

class Builder extends AnyRef

Linear Supertypes
AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. Builder
  2. AnyRef
  3. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

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(): FeedForward

    Permalink

    Construct the FeedForward model and fit on the input training data

    Construct the FeedForward model and fit on the input training data

    returns

    the trained model

  6. def clone(): AnyRef

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

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

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

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

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

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

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

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

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

    Permalink
    Definition Classes
    AnyRef
  16. def setAllowExtraParams(allowExtraParams: Boolean): Builder

    Permalink

    Whether allow extra parameters that are not needed by symbol to be passed by aux_params and arg_params.

    Whether allow extra parameters that are not needed by symbol to be passed by aux_params and arg_params. If this is True, no error will be thrown when aux_params and arg_params contain extra parameters than needed.

  17. def setArgParams(argParams: Map[String, NDArray]): Builder

    Permalink

    Set the model parameter, dict of name to NDArray of net's weights.

  18. def setAuxParams(auxParams: Map[String, NDArray]): Builder

    Permalink

    Set the model parameter, dict of name to NDArray of net's auxiliary states

  19. def setBatchEndCallback(batchEndCallback: BatchEndCallback): Builder

    Permalink

    batchEndCallback A callback that is invoked at end of each batch.

    batchEndCallback A callback that is invoked at end of each batch. For print purpose.

  20. def setBatchSize(batchSize: Int): Builder

    Permalink

    Set the batch size of training data.

  21. def setBeginEpoch(beginEpoch: Int): Builder

    Permalink

    Set the beginning training epoch.

  22. def setContext(ctx: Array[Context]): Builder

    Permalink

    Set ctx The device context of training and prediction.

    Set ctx The device context of training and prediction. To use multi GPU training, pass in a list of gpu contexts.

  23. def setEpochEndCallback(epochEndCallback: EpochEndCallback): Builder

    Permalink

    A callback that is invoked at end of each epoch.

    A callback that is invoked at end of each epoch. This can be used to checkpoint model each epoch.

  24. def setEpochSize(epochSize: Int): Builder

    Permalink

    Set number of batches in a epoch.

    Set number of batches in a epoch. In default, it is set to ceil(num_train_examples / batch_size)

  25. def setEvalData(evalData: DataIter): Builder

    Permalink

    Set the evaluation data

  26. def setEvalMetric(metric: EvalMetric): Builder

    Permalink

    Set the evaluation metric.

    Set the evaluation metric. Default Accuracy()

  27. def setInitializer(initializer: Initializer): Builder

    Permalink

    Set the initialization scheme used.

    Set the initialization scheme used. Default Uniform(0.01f).

  28. def setKVStore(kv: String): Builder

    Permalink

    A string kvstore type: 'local' : multi-devices on a single machine, will automatically choose one from 'local_update_cpu', 'local_allreduce_cpu', and 'local_allreduce_device' 'dist_sync' : multi-machines with BSP 'dist_async' : multi-machines with partical asynchronous In default uses 'local', often no need to change for single machine.

  29. def setKVStore(kv: KVStore): Builder

    Permalink

    this will take precedence over the setKVStore(String) version

  30. def setLogger(logger: Logger): Builder

    Permalink

    When not specified, default logger will be used.

  31. def setNumEpoch(numEpoch: Int): Builder

    Permalink

    Set number of training epochs

  32. def setOptimizer(opt: Optimizer): Builder

    Permalink

    Set optimizer for training.

    Set optimizer for training. Default SGD.

  33. def setTrainData(trainData: DataIter): Builder

    Permalink

    Set the training data

  34. def setWorkLoadList(workLoadList: Seq[Float]): Builder

    Permalink

    Set the list of work load for different devices, in the same order as ctx

  35. def setup(): FeedForward

    Permalink

    Construct the FeedForward model but do NOT train

    Construct the FeedForward model but do NOT train

    returns

    the un-trained model

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

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

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

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

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

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

Inherited from AnyRef

Inherited from Any

Ungrouped