Package

ml.dmlc

mxnet

Permalink

package mxnet

Visibility
  1. Public
  2. All

Type Members

  1. class Accuracy extends EvalMetric

    Permalink

    Calculate accuracy

  2. trait BatchEndCallback extends AnyRef

    Permalink
  3. class CompositeEvalMetric extends EvalMetric

    Permalink

    Manage multiple evaluation metrics.

  4. class Context extends Serializable

    Permalink

    Constructing a context.

  5. class CustomMetric extends EvalMetric

    Permalink

    Custom evaluation metric that takes a NDArray function.

  6. abstract class CustomOp extends AnyRef

    Permalink

    Base class for operators implemented in Scala

  7. abstract class CustomOpProp extends AnyRef

    Permalink

    Base class for operator property class implemented in Scala.

    Base class for operator property class implemented in Scala. MXNET_CPU_WORKER_NTHREADS must be greater than 1 for custom op to work on CPU

  8. class DataBatch extends AnyRef

    Permalink

    class batch of data

  9. case class DataDesc(name: String, shape: Shape, dtype: DType = Base.MX_REAL_TYPE, layout: String = "NCHW") extends Product with Serializable

    Permalink
  10. abstract class DataIter extends Iterator[DataBatch]

    Permalink

    DataIter object in mxnet.

  11. abstract class DataPack extends Iterable[DataBatch]

    Permalink

    pack of DataIter, use as Iterable class

  12. trait EpochEndCallback extends AnyRef

    Permalink
  13. abstract class EvalMetric extends AnyRef

    Permalink

    Base class of all evaluation metrics

  14. class Executor extends AnyRef

    Permalink

    Symbolic Executor component of MXNet
    WARNING: it is your responsibility to clear this object through dispose(). NEVER rely on the GC strategy

Symbolic Executor component of MXNet
WARNING: it is your responsibility to clear this object through dispose(). NEVER rely on the GC strategy

See also

Symbol.bind : to create executor

  • class F1 extends EvalMetric

    Permalink

    Calculate the F1 score of a binary classification problem.

  • class FactorScheduler extends LRScheduler

    Permalink

    Class for reducing learning rate in factor

    Class for reducing learning rate in factor

    Assume the weight has been updated by n times, then the learning rate will be base_lr * factor^^(floor(n/step))

  • class FeedForward extends AnyRef

    Permalink

    Model class of MXNet for training and predicting feedforward nets.

    Model class of MXNet for training and predicting feedforward nets. This class is designed for a single-data single output supervised network.

  • abstract class Initializer extends AnyRef

    Permalink

    Base class for Initializer.

  • class KVStore extends AnyRef

    Permalink
  • abstract class LRScheduler extends AnyRef

    Permalink

    Learning rate scheduler, which adaptively changes the learning rate based on the training progress.

  • class MAE extends EvalMetric

    Permalink

    Calculate Mean Absolute Error loss

  • class MSE extends EvalMetric

    Permalink
  • class MXIndexedRecordIO extends MXRecordIO

    Permalink

    Scala interface for read/write RecordIO data formmat with index.

    Scala interface for read/write RecordIO data formmat with index. Support random access.

  • trait MXKVStoreCachedStates extends AnyRef

    Permalink
  • trait MXKVStoreUpdater extends AnyRef

    Permalink
  • class MXRecordIO extends AnyRef

    Permalink

    Scala interface for read/write RecordIO data format

  • class Mixed extends Initializer

    Permalink

    Initialize the weight with mixed Initializer

  • class Model extends AnyRef

    Permalink

    Describe the model flow

  • class Monitor extends AnyRef

    Permalink

    Monitor outputs, weights, and gradients for debugging.

  • class NDArray extends AnyRef

    Permalink

    NDArray object in mxnet.

    NDArray object in mxnet. NDArray is basic ndarray/Tensor like data structure in mxnet.
    WARNING: it is your responsibility to clear this object through dispose(). NEVER rely on the GC strategy

  • class NameManager extends AnyRef

    Permalink

    NameManager to do automatic naming.

    NameManager to do automatic naming. User can also inherit this object to change naming behavior.

  • class Normal extends Initializer

    Permalink

    Initialize the weight with normal(0, sigma)

  • abstract class Optimizer extends Serializable

    Permalink
  • class Perplexity extends EvalMetric

    Permalink

    Calculate perplexity.

  • class RMSE extends EvalMetric

    Permalink

    Calculate Root Mean Squred Error loss

  • class Rtc extends AnyRef

    Permalink

    This class allow you to write cuda kernel in Scala and call them with NDArray.

  • class Shape extends Serializable

    Permalink

    Shape of NDArray or other data

  • class Symbol extends AnyRef

    Permalink

    Symbolic configuration API of mxnet.

    Symbolic configuration API of mxnet.
    WARNING: it is your responsibility to clear this object through dispose(). NEVER rely on the GC strategy

  • class SymbolConversions[V] extends AnyRef

    Permalink
  • trait SymbolGenerator extends AnyRef

    Permalink
  • class TopKAccuracy extends EvalMetric

    Permalink

    Calculate top k predictions accuracy

  • class Uniform extends Initializer

    Permalink

    Initialize the weight with uniform [-scale, scale]

  • class Xavier extends Initializer

    Permalink

    Initialize the weight with Xavier or similar initialization scheme.

  • Value Members

    1. object Callback

      Permalink

      Callback functions that can be used to track various status during epoch.

    2. object Context extends Serializable

      Permalink
    3. object DType extends Enumeration

      Permalink
    4. object DataDesc extends Serializable

      Permalink
    5. object Executor

      Permalink
    6. object FeedForward

      Permalink
    7. object IO

      Permalink

      IO iterators for loading training & validation data

    8. object KVStore

      Permalink

      Key value store interface of MXNet for parameter synchronization.

    9. object KVStoreServer

      Permalink
    10. object MXRecordIO

      Permalink
    11. object Model

      Permalink
    12. object NDArray

      Permalink

      NDArray API of mxnet

    13. object NameManager

      Permalink
    14. object Operator

      Permalink
    15. object Optimizer extends Serializable

      Permalink
    16. object Profiler

      Permalink
    17. object Random

      Permalink

      Random Number interface of mxnet.

    18. object Shape extends Serializable

      Permalink
    19. object Symbol

      Permalink
    20. object SymbolConversions

      Permalink
    21. object Visualization

      Permalink
    22. package contrib

      Permalink
    23. package io

      Permalink
    24. package module

      Permalink
    25. package optimizer

      Permalink

    Ungrouped