ZooModel

abstract def buildModel(): AbstractModule[A, B, T]

Override this method to define a model.

final def !=(arg0: AnyRef): Boolean

final def !=(arg0: Any): Boolean

final def ##(): Int

final def ==(arg0: AnyRef): Boolean

final def ==(arg0: Any): Boolean

def accGradParameters(input: A, gradOutput: B): Unit

def addModel(model: AbstractModule[A, B, T]): ZooModel.this.type

def apply(name: String): Option[AbstractModule[Activity, Activity, T]]

final def asInstanceOf[T0]: T0

def backward(input: A, gradOutput: B): A

var backwardTime: Long

def build(): ZooModel.this.type

def canEqual(other: Any): Boolean

final def checkEngineType(): ZooModel.this.type

def clearState(): ZooModel.this.type

final def clone(deepCopy: Boolean): AbstractModule[A, B, T]

def clone(): AnyRef

final def cloneModule(): AbstractModule[A, B, T]

final def eq(arg0: AnyRef): Boolean

def equals(other: Any): Boolean

final def evaluate(): ZooModel.this.type

final def evaluate(dataSet: LocalDataSet[MiniBatch[T]], vMethods: Array[_ <: ValidationMethod[T]]): Array[(ValidationResult, ValidationMethod[T])]

final def evaluate(dataset: RDD[Sample[T]], vMethods: Array[_ <: ValidationMethod[T]], batchSize: Option[Int]): Array[(ValidationResult, ValidationMethod[T])]

final def evaluateImage(imageFrame: ImageFrame, vMethods: Array[_ <: ValidationMethod[T]], batchSize: Option[Int]): Array[(ValidationResult, ValidationMethod[T])]

def finalize(): Unit

def findModules(moduleType: String): ArrayBuffer[AbstractModule[_, _, T]]

final def forward(input: A): B

var forwardTime: Long

def freeze(names: String*): ZooModel.this.type

final def getClass(): Class[_]

def getExtraParameter(): Array[Tensor[T]]

final def getInputShape(): Shape

final def getName(): String

final def getNumericType(): TensorDataType

final def getOutputShape(): Shape

def getParametersTable(): Table

final def getPrintName(): String

final def getScaleB(): Double

final def getScaleW(): Double

def getTimes(): Array[(AbstractModule[_ <: Activity, _ <: Activity, T], Long, Long)]

final def getTimesGroupByModuleType(): Array[(String, Long, Long)]

final def getWeightsBias(): Array[Tensor[T]]

var gradInput: A

final def hasName: Boolean

def hashCode(): Int

def inputs(first: (ModuleNode[T], Int), nodesWithIndex: (ModuleNode[T], Int)*): ModuleNode[T]

def inputs(nodes: Array[ModuleNode[T]]): ModuleNode[T]

def inputs(nodes: ModuleNode[T]*): ModuleNode[T]

final def isInstanceOf[T0]: Boolean

final def isTraining(): Boolean

var line: String

final def loadModelWeights(srcModel: Module[Float], matchAll: Boolean): ZooModel.this.type

final def loadWeights(weightPath: String, matchAll: Boolean): ZooModel.this.type

def model: AbstractModule[A, B, T]

The defined model, either from buildModel() or loaded from file.

val modules: ArrayBuffer[AbstractModule[Activity, Activity, T]]

final def ne(arg0: AnyRef): Boolean

final def notify(): Unit

final def notifyAll(): Unit

var output: B

def parameters(): (Array[Tensor[T]], Array[Tensor[T]])

final def predict(dataset: RDD[Sample[T]], batchSize: Int, shareBuffer: Boolean): RDD[Activity]

final def predictClass(dataset: RDD[Sample[T]], batchSize: Int): RDD[Int]

def predictClasses(x: RDD[Sample[T]], batchSize: Int = 1, zeroBasedLabel: Boolean = true): RDD[Int]

Predict for classes.

final def predictImage(imageFrame: ImageFrame, outputLayer: String, shareBuffer: Boolean, batchPerPartition: Int, predictKey: String, featurePaddingParam: Option[PaddingParam[T]]): ImageFrame

def processInputs(first: (ModuleNode[T], Int), nodesWithIndex: (ModuleNode[T], Int)*): ModuleNode[T]

def processInputs(nodes: Seq[ModuleNode[T]]): ModuleNode[T]

final def quantize(): Module[T]

def release(): Unit

def reset(): Unit

def resetTimes(): Unit

final def saveCaffe(prototxtPath: String, modelPath: String, useV2: Boolean, overwrite: Boolean): ZooModel.this.type

final def saveDefinition(path: String, overWrite: Boolean): ZooModel.this.type

def saveModel(path: String, weightPath: String = null, overWrite: Boolean = false): ZooModel.this.type

Save the model to the specified path.

final def saveModule(path: String, weightPath: String, overWrite: Boolean): ZooModel.this.type

final def saveTF(inputs: Seq[(String, Seq[Int])], path: String, byteOrder: ByteOrder, dataFormat: TensorflowDataFormat): ZooModel.this.type

final def saveTorch(path: String, overWrite: Boolean): ZooModel.this.type

final def saveWeights(path: String, overWrite: Boolean): Unit

var scaleB: Double

var scaleW: Double

final def setExtraParameter(extraParam: Array[Tensor[T]]): ZooModel.this.type

final def setLine(line: String): ZooModel.this.type

final def setName(name: String): ZooModel.this.type

def setScaleB(b: Double): ZooModel.this.type

def setScaleW(w: Double): ZooModel.this.type

final def setWeightsBias(newWeights: Array[Tensor[T]]): ZooModel.this.type

def summary(): Unit

Print out the summary of the model.

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

def toGraph(startNodes: ModuleNode[T]*): Graph[T]

def toString(): String

var train: Boolean

final def training(): ZooModel.this.type

def unFreeze(names: String*): ZooModel.this.type

def updateGradInput(input: A, gradOutput: B): A

def updateOutput(input: A): B

final def wait(): Unit

final def wait(arg0: Long, arg1: Int): Unit

final def wait(arg0: Long): Unit

def zeroGradParameters(): Unit

final def save(path: String, overWrite: Boolean): ZooModel.this.type