mxnet

Type Members

class Accuracy extends EvalMetric

Calculate accuracy
trait BatchEndCallback extends AnyRef
class CompositeEvalMetric extends EvalMetric

Manage multiple evaluation metrics.
class Context extends Serializable

Constructing a context.
class CustomMetric extends EvalMetric

Custom evaluation metric that takes a NDArray function.
abstract class CustomOp extends AnyRef

Base class for operators implemented in Scala
abstract class CustomOpProp extends AnyRef

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
class DataBatch extends AnyRef

class batch of data
case class DataDesc(name: String, shape: Shape, dtype: DType.DType = DType.Float32, layout: String = Layout.UNDEFINED) extends Product with Serializable
abstract class DataIter extends Iterator[DataBatch]

DataIter object in mxnet.
abstract class DataPack extends Iterable[DataBatch]

pack of DataIter, use as Iterable class
trait EpochEndCallback extends AnyRef
abstract class EvalMetric extends AnyRef

Base class of all evaluation metrics
class Executor extends NativeResource

Symbolic Executor component of MXNet
WARNING: it is your responsibility to clear this object through dispose().

Value Members

object Callback

Callback functions that can be used to track various status during epoch.
object CheckUtils
object Context extends Serializable
object DType extends Enumeration
object DataBatch
object DataDesc extends Serializable
object Executor
object FeedForward
object IO

IO iterators for loading training & validation data
object Image

Image API of Scala package enable OpenCV feature
object KVStore

Key value store interface of MXNet for parameter synchronization.
object KVStoreServer
object Layout

Layout definition of DataDesc N Batch size C channels H Height W Weight T sequence length undefined default value of Layout
object MXRecordIO
object Model
object NDArray extends NDArrayBase

NDArray Object extends from NDArrayBase for abstract function signatures Main code will be generated during compile time through Macros
NDArray Object extends from NDArrayBase for abstract function signatures Main code will be generated during compile time through Macros

Annotations
@AddNDArrayFunctions( false )
object NDArrayAPI extends NDArrayAPIBase

Annotations
@AddNDArrayAPIs( false )
object NDArrayCollector

A collector to store NDArrays.
A collector to store NDArrays. It provides a scope, NDArrays allocated in the scope can either
- be disposed automatically when the code block finishes, or
- simply be collected for future usage.
  If the return type of scope is NDArray or NDArrayFuncReturn, the collector is smart enough NOT to collect or dispose the returned NDArray.
  However in other cases, it is users' responsibility NOT to leak allocated NDArrays outside, (e.g., store to a global variable and use later, pass to another thread, etc.)
  Usage Example:
```
 val a = NDArray.array(Array(-1f, 0f, 1f, 2f, 3f, 4f), shape = Shape(2, 3))
 val res = NDArrayCollector.auto().withScope {
   (NDArray.relu(a) + a).toArray
 }
```
  In the case above, the intermediate NDArrays (created by NDArray.relu and +) will be disposed automatically.
  User can also decide to dispose the collected NDArrays later:
```
 val collector = NDArrayCollector.manual()
 val res = collector.withScope {
   (NDArray.relu(a) + a).toArray
 }
 collector.foreach(_.dispose())
```
  For Java users:
```
 NDArray a = NDArray.array(new float[]{-1f, 0f, 1f, 2f, 3f, 4f},
                           Shape.create(2, 3), Context.cpu(0));
 float[] sliced = NDArrayCollector.auto().withScope(
   new scala.runtime.AbstractFunction0() {
   @Override
   public float[] apply() {
     a.slice(0, 1).toArray();
   }
 });
```
object NameManager
object Operator
object Optimizer extends Serializable
object Profiler
object Random

Random Number interface of mxnet.
object ResourceScope
object Shape extends Serializable
object Symbol extends SymbolBase

Symbol Object extends from SymbolBase for abstract function signatures Main code will be generated during compile time through Macros
Symbol Object extends from SymbolBase for abstract function signatures Main code will be generated during compile time through Macros

Annotations
@AddSymbolFunctions( false )
object SymbolAPI extends SymbolAPIBase

Annotations
@AddSymbolAPIs( false )
object SymbolConversions
object Visualization
package annotation
package contrib
package infer
package io
package module
package optimizer

package mxnet

Type Members

class Accuracy extends EvalMetric

trait BatchEndCallback extends AnyRef

class CompositeEvalMetric extends EvalMetric

class Context extends Serializable

class CustomMetric extends EvalMetric

abstract class CustomOp extends AnyRef

abstract class CustomOpProp extends AnyRef

class DataBatch extends AnyRef

case class DataDesc(name: String, shape: Shape, dtype: DType.DType = DType.Float32, layout: String = Layout.UNDEFINED) extends Product with Serializable

abstract class DataIter extends Iterator[DataBatch]

abstract class DataPack extends Iterable[DataBatch]

trait EpochEndCallback extends AnyRef

abstract class EvalMetric extends AnyRef

class Executor extends NativeResource

class F1 extends EvalMetric

class FactorScheduler extends LRScheduler

class FeedForward extends NativeResource

abstract class Initializer extends AnyRef

class KVStore extends NativeResource

abstract class LRScheduler extends AnyRef

class MAE extends EvalMetric

class MSE extends EvalMetric

class MXIndexedRecordIO extends MXRecordIO

trait MXKVStoreCachedStates extends AnyRef

trait MXKVStoreUpdater extends NativeResource

class MXNetError extends Exception

class MXRecordIO extends AnyRef

class Mixed extends Initializer

class Model extends AnyRef

class Monitor extends AnyRef

class NDArray extends NativeResource

abstract class NDArrayAPIBase extends AnyRef

abstract class NDArrayBase extends AnyRef

class NDArrayCollector extends AnyRef

class NameManager extends AnyRef

class Normal extends Initializer

abstract class Optimizer extends Serializable

class Perplexity extends EvalMetric

class RMSE extends EvalMetric

class ResourceScope extends AutoCloseable

class Rtc extends AnyRef

class Shape extends Serializable

class Symbol extends NativeResource

abstract class SymbolAPIBase extends AnyRef

abstract class SymbolBase extends AnyRef

class SymbolConversions[V] extends AnyRef

trait SymbolGenerator extends AnyRef

class TestUtil extends AnyRef

class TopKAccuracy extends EvalMetric

class Uniform extends Initializer

trait WarnIfNotDisposed extends AnyRef

class Xavier extends Initializer

Value Members

object Callback

object CheckUtils

object Context extends Serializable

object DType extends Enumeration

object DataBatch

object DataDesc extends Serializable

object Executor

object FeedForward

object IO

object Image

object KVStore

object KVStoreServer

object Layout

object MXRecordIO

object Model

object NDArray extends NDArrayBase

object NDArrayAPI extends NDArrayAPIBase

object NDArrayCollector

object NameManager

object Operator

object Optimizer extends Serializable

object Profiler

object Random

object ResourceScope