Trait

org.deeplearning4j.scalnet.models

Model

Related Doc: package models

Permalink

trait Model extends Logging

Abstract base class for neural net architectures.

Linear Supertypes
Logging, AnyRef, Any
Known Subclasses
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. Model
  2. Logging
  3. AnyRef
  4. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Abstract Value Members

  1. abstract def compile(lossFunction: LossFunction, optimizer: OptimizationAlgorithm, updater: Updater): Unit

    Permalink

    Compile neural net architecture.

    Compile neural net architecture. Call immediately before training.

    lossFunction

    loss function to use

    optimizer

    optimization algorithm to use

Concrete 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 buildModelConfig(optimizer: OptimizationAlgorithm, updater: Updater, miniBatch: Boolean, biasInit: Double, seed: Long): Builder

    Permalink

    Build model configuration from optimizer and seed.

    Build model configuration from optimizer and seed.

    optimizer

    optimization algorithm to use in model

    seed

    seed to use

    returns

    NeuralNetConfiguration.Builder

  6. def buildOutput(lossFunction: LossFunction): Unit

    Permalink

    Make last layer of architecture an output layer using the provided loss function.

    Make last layer of architecture an output layer using the provided loss function.

    lossFunction

    loss function to use

  7. def clone(): AnyRef

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

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

    Permalink
    Definition Classes
    AnyRef → Any
  10. def evaluate(dataset: DataSet, numClasses: Int): Evaluation

    Permalink

    Evaluate model against an iterator over data set

    Evaluate model against an iterator over data set

    dataset

    data set

    numClasses

    output size

    returns

    Evaluation instance

  11. def evaluate(dataset: DataSet): Evaluation

    Permalink

    Evaluate model against an iterator over data set

    Evaluate model against an iterator over data set

    dataset

    data set

    returns

    Evaluation instance

  12. def evaluate(iter: DataSetIterator, numClasses: Int): Evaluation

    Permalink

    Evaluate model against an iterator over data set

    Evaluate model against an iterator over data set

    iter

    iterator over data set

    numClasses

    output size

    returns

    Evaluation instance

  13. def evaluate(iter: DataSetIterator): Evaluation

    Permalink

    Evaluate model against an iterator over data set

    Evaluate model against an iterator over data set

    iter

    iterator over data set

    returns

    Evaluation instance

  14. def finalize(): Unit

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  15. def fit(dataset: DataSet, nbEpoch: Int, listeners: List[IterationListener]): Unit

    Permalink

    Fit neural net to data.

    Fit neural net to data.

    dataset

    data set

    nbEpoch

    number of epochs to train

    listeners

    callbacks for monitoring training

  16. def fit(iter: DataSetIterator, nbEpoch: Int, listeners: List[IterationListener]): Unit

    Permalink

    Fit neural net to data.

    Fit neural net to data.

    iter

    iterator over data set

    nbEpoch

    number of epochs to train

    listeners

    callbacks for monitoring training

  17. final def getClass(): Class[_]

    Permalink
    Definition Classes
    AnyRef → Any
  18. def getLayers: List[Node]

    Permalink
  19. def getNetwork: MultiLayerNetwork

    Permalink
  20. def hashCode(): Int

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

    Permalink
    Definition Classes
    Any
  22. var layers: List[Node]

    Permalink
    Attributes
    protected
  23. lazy val logger: Logger

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  24. var model: MultiLayerNetwork

    Permalink
    Attributes
    protected
  25. final def ne(arg0: AnyRef): Boolean

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

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

    Permalink
    Definition Classes
    AnyRef
  28. def predict(x: DataSet): INDArray

    Permalink

    Use neural net to make prediction on input x.

    Use neural net to make prediction on input x.

    x

    input represented as DataSet

  29. def predict(x: INDArray): INDArray

    Permalink

    Use neural net to make prediction on input x

    Use neural net to make prediction on input x

    x

    input represented as INDArray

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

    Permalink
    Definition Classes
    AnyRef
  31. def toJson: String

    Permalink
  32. def toString(): String

    Permalink
    Definition Classes
    Model → AnyRef → Any
  33. def toYaml: String

    Permalink
  34. final def wait(): Unit

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

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

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

Inherited from Logging

Inherited from AnyRef

Inherited from Any

Ungrouped