Manage multiple evaluation metrics.
Constructing a context.
Custom evaluation metric that takes a NDArray function.
Base class for operators implemented in Scala
Base class for operator property class implemented in Scala.
class batch of data
DataIter object in mxnet.
pack of DataIter, use as Iterable class
Base class of all evaluation metrics
Symbolic Executor component of MXNet
WARNING: it is your responsibility to clear this object through dispose().
Symbol.bind : to create executor
Calculate the F1 score of a binary classification problem.
Class for reducing learning rate in factor
Model class of MXNet for training and predicting feedforward nets.
Base class for Initializer.
Learning rate scheduler, which adaptively changes the learning rate based on the training progress.
Calculate Mean Absolute Error loss
Scala interface for read/write RecordIO data formmat with index.
Scala interface for read/write RecordIO data format
Initialize the weight with mixed Initializer
Describe the model flow
Monitor outputs, weights, and gradients for debugging.
NDArray object in mxnet.
NameManager to do automatic naming.
Initialize the weight with normal(0, sigma)
Calculate Root Mean Squred Error loss
This class manages automatically releasing of
This class allow you to write cuda kernel in Scala and call them with NDArray.
Shape of NDArray or other data
Symbolic configuration API of mxnet.
Calculate top k predictions accuracy
Initialize the weight with uniform [-scale, scale]
Initialize the weight with Xavier or similar initialization scheme.
Callback functions that can be used to track various status during epoch.
IO iterators for loading training & validation data
Image API of Scala package enable OpenCV feature
Key value store interface of MXNet for parameter synchronization.
Layout definition of DataDesc N Batch size C channels H Height W Weight T sequence length undefined default value of Layout
NDArray Object extends from NDArrayBase for abstract function signatures Main code will be generated during compile time through Macros
A collector to store NDArrays.
Random Number interface of mxnet.
Symbol Object extends from SymbolBase for abstract function signatures Main code will be generated during compile time through Macros