KerasZooModel

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

Override this method to define a model.

final def !=(arg0: Any): Boolean

final def ##(): Int

final def ==(arg0: Any): Boolean

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

def addModel(model: AbstractModule[A, B, T]): KerasZooModel.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(): KerasZooModel.this.type

def canEqual(other: Any): Boolean

final def checkEngineType(): KerasZooModel.this.type

def clearGradientClipping(): Unit

def clearState(): KerasZooModel.this.type

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

def clone(): AnyRef

final def cloneModule(): KerasZooModel.this.type

def compile(optimizer: OptimMethod[T], loss: (Variable[T], Variable[T]) ⇒ Variable[T])(implicit ev: TensorNumeric[T]): Unit

def compile(optimizer: OptimMethod[T], loss: (Variable[T], Variable[T]) ⇒ Variable[T], metrics: List[ValidationMethod[T]])(implicit ev: TensorNumeric[T]): Unit

def compile(optimizer: String, loss: String)(implicit ev: TensorNumeric[T]): Unit

def compile(optimizer: String, loss: String, metrics: List[String])(implicit ev: TensorNumeric[T]): Unit

def compile(optimizer: OptimMethod[T], loss: Criterion[T], metrics: List[ValidationMethod[T]] = null)(implicit ev: TensorNumeric[T]): Unit

final def eq(arg0: AnyRef): Boolean

def equals(other: Any): Boolean

def evaluate(x: TextSet, batchSize: Int): Array[(ValidationResult, ValidationMethod[T])]

def evaluate(x: ImageSet, batchSize: Int)(implicit ev: TensorNumeric[T]): Array[(ValidationResult, ValidationMethod[T])]

def evaluate(x: LocalDataSet[MiniBatch[T]])(implicit ev: TensorNumeric[T]): Array[(ValidationResult, ValidationMethod[T])]

def evaluate(x: RDD[Sample[T]], batchSize: Int)(implicit ev: TensorNumeric[T]): Array[(ValidationResult, ValidationMethod[T])]

final def evaluate(): KerasZooModel.this.type

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

final def evaluate(dataset: RDD[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]]

def fit(x: TextSet, batchSize: Int, nbEpoch: Int)(implicit ev: TensorNumeric[T]): Unit

def fit(x: TextSet, batchSize: Int, nbEpoch: Int, validationData: TextSet)(implicit ev: TensorNumeric[T]): Unit

def fit(x: ImageSet, batchSize: Int, nbEpoch: Int)(implicit ev: TensorNumeric[T]): Unit

def fit(x: ImageSet, batchSize: Int, nbEpoch: Int, validationData: ImageSet)(implicit ev: TensorNumeric[T]): Unit

def fit(x: RDD[Sample[T]], batchSize: Int = 32, nbEpoch: Int = 10, validationData: RDD[Sample[T]] = null, featurePaddingParam: PaddingParam[T] = null, labelPaddingParam: PaddingParam[T] = null)(implicit ev: TensorNumeric[T]): Unit

def fit(x: DataSet[MiniBatch[T]], nbEpoch: Int)(implicit ev: TensorNumeric[T]): Unit

def fit(x: DataSet[MiniBatch[T]], nbEpoch: Int, validationData: DataSet[MiniBatch[T]])(implicit ev: TensorNumeric[T]): Unit

final def forward(input: A): B

var forwardTime: Long

def freeze(names: String*): KerasZooModel.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)]

def getTrainSummary(tag: String): Array[(Long, Float, Double)]

def getValidationSummary(tag: String): Array[(Long, Float, Double)]

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): KerasZooModel.this.type

final def loadWeights(weightPath: String, matchAll: Boolean): KerasZooModel.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]])

def predict(x: RDD[Sample[T]], batchPerThread: Int): RDD[Activity]

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): KerasZooModel.this.type

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

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

Save the model to the specified path.

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

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

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

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

var scaleB: Double

var scaleW: Double

def setCheckpoint(path: String, overWrite: Boolean = true): Unit

def setConstantGradientClipping(min: Float, max: Float): Unit

def setEvaluateStatus(): KerasZooModel.this.type

Set the model to be in evaluate status, i.e.

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

def setGradientClippingByL2Norm(clipNorm: Float): Unit

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

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

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

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

def setTensorBoard(logDir: String, appName: String): Unit

final def setWeightsBias(newWeights: Array[Tensor[T]]): KerasZooModel.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(): KerasZooModel.this.type

def unFreeze(names: String*): KerasZooModel.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): KerasZooModel.this.type