org.apache.mxnet.infer

Classifier

Related Doc: package infer

class Classifier extends ClassifierBase

A class for classifier tasks

Linear Supertypes
ClassifierBase, AnyRef, Any
Known Subclasses
Ordering
  1. Alphabetic
  2. By inheritance
Inherited
  1. Classifier
  2. ClassifierBase
  3. AnyRef
  4. Any
  1. Hide All
  2. Show all
Learn more about member selection
Visibility
  1. Public
  2. All

Instance Constructors

  1. new Classifier(modelPathPrefix: String, inputDescriptors: IndexedSeq[DataDesc], contexts: Array[Context] = Context.cpu(), epoch: Option[Int] = Some(0))

    modelPathPrefix

    Path prefix from where to load the model artifacts These include the symbol, parameters, and synset.txt Example: file://model-dir/resnet-152 (containing resnet-152-symbol.json, resnet-152-0000.params, and synset.txt)

    inputDescriptors

    Descriptors defining the input node names, shape, layout and type parameters

    contexts

    Device contexts on which you want to run inference; defaults to CPU

    epoch

    Model epoch to load; defaults to 0

Value Members

  1. final def !=(arg0: Any): Boolean

    Definition Classes
    AnyRef → Any
  2. final def ##(): Int

    Definition Classes
    AnyRef → Any
  3. final def ==(arg0: Any): Boolean

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

    Definition Classes
    Any
  5. def classify[T](input: IndexedSeq[Array[T]], topK: Option[Int] = None): IndexedSeq[(String, T)]

    Takes flat arrays as input and returns (Label, Score) tuples.

    Takes flat arrays as input and returns (Label, Score) tuples.

    T

    The Scala equivalent of the DType used for the input array and return value

    input

    Indexed sequence one-dimensional array of floats/doubles

    topK

    (Optional) How many result (sorting based on the last axis) elements to return. Default returns unsorted output.

    returns

    Indexed sequence of (Label, Score) tuples

    Definition Classes
    ClassifierClassifierBase
  6. def classifyWithNDArray(input: IndexedSeq[NDArray], topK: Option[Int] = None): IndexedSeq[IndexedSeq[(String, Float)]]

    Perform multiple classification operations on NDArrays.

    Perform multiple classification operations on NDArrays. Also works with batched input.

    input

    Indexed sequence of NDArrays

    topK

    (Optional) How many result (sorting based on the last axis) elements to return. Default returns unsorted output.

    returns

    Traversable sequence of (Label, Score) tuples.

    Definition Classes
    ClassifierClassifierBase
  7. def clone(): AnyRef

    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  8. val contexts: Array[Context]

    Device contexts on which you want to run inference; defaults to CPU

    Device contexts on which you want to run inference; defaults to CPU

    Attributes
    protected
  9. val epoch: Option[Int]

    Model epoch to load; defaults to 0

    Model epoch to load; defaults to 0

    Attributes
    protected
  10. final def eq(arg0: AnyRef): Boolean

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

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

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

    Definition Classes
    AnyRef → Any
  14. val handler: MXNetHandler

    Attributes
    protected[org.apache.mxnet.infer]
  15. def hashCode(): Int

    Definition Classes
    AnyRef → Any
  16. val inputDescriptors: IndexedSeq[DataDesc]

    Descriptors defining the input node names, shape, layout and type parameters

    Descriptors defining the input node names, shape, layout and type parameters

    Attributes
    protected
  17. final def isInstanceOf[T0]: Boolean

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

    Definition Classes
    AnyRef
  19. final def notify(): Unit

    Definition Classes
    AnyRef
  20. final def notifyAll(): Unit

    Definition Classes
    AnyRef
  21. val predictor: PredictBase

    Attributes
    protected[org.apache.mxnet.infer]
  22. final def synchronized[T0](arg0: ⇒ T0): T0

    Definition Classes
    AnyRef
  23. val synset: IndexedSeq[String]

    Attributes
    protected[org.apache.mxnet.infer]
  24. val synsetFilePath: String

    Attributes
    protected[org.apache.mxnet.infer]
  25. def toString(): String

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

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

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

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

Inherited from ClassifierBase

Inherited from AnyRef

Inherited from Any

Ungrouped