GraphNet

final def !=(arg0: Any): Boolean

final def ##(): Int

final def ==(arg0: Any): Boolean

def accGradParameters(input: Activity, gradOutput: Activity): Unit

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

final def asInstanceOf[T0]: T0

def backward(input: Activity, gradOutput: Activity): Activity

var backwardTime: Long

def canEqual(other: Any): Boolean

final def checkEngineType(): GraphNet.this.type

def clearState(): GraphNet.this.type

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

def clone(): AnyRef

final def cloneModule(): GraphNet.this.type

final def eq(arg0: AnyRef): Boolean

def equals(other: Any): Boolean

implicit val ev: TensorNumeric[T]

final def evaluate(): GraphNet.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]]

final def forward(input: Activity): Activity

var forwardTime: Long

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

def freezeUpTo(names: String*): GraphNet.this.type

Freeze the model from the bottom up to the layers specified by names (inclusive).

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 getSubModules(): List[AbstractModule[Activity, Activity, T]]

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

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

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

var gradInput: Activity

final def hasName: Boolean

def hashCode(): Int

val inputNodes: Seq[ModuleNode[T]]

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

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

val module: Module[T]

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

final def ne(arg0: AnyRef): Boolean

def newGraph(outputs: Seq[String]): GraphNet[T]

Specify a seq of nodes as output and return a new graph using the existing nodes

def newGraph(output: String): GraphNet[T]

Specify a node as output and return a new graph using the existing nodes

def node(name: String): ModuleNode[T]

Return the node in the graph as specified by the name

def nodes(names: Seq[String]): Seq[ModuleNode[T]]

Return the nodes in the graph as specified by the names

final def notify(): Unit

final def notifyAll(): Unit

var output: Activity

val outputNodes: Seq[ModuleNode[T]]

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

def predict(x: TextSet): TextSet

The default batchPerThread is 4.

def predict(x: TextSet, batchPerThread: Int): TextSet

Use a model to do prediction on TextSet.

def predict(x: ImageSet): ImageSet

The default batchPerThread is 4.

def predict(x: ImageSet, batchPerThread: Int): ImageSet

Use a model to do prediction on ImageSet.

def predict(x: Array[Sample[T]])(implicit ev: TensorNumeric[T]): Array[Activity]

Use a model to do prediction in local mode.

def predict(x: Array[Sample[T]], batchPerThread: Int)(implicit ev: TensorNumeric[T]): Array[Activity]

Use a model to do prediction in local mode.

def predict(x: LocalDataSet[MiniBatch[T]])(implicit ev: TensorNumeric[T]): Array[Activity]

Use a model to do prediction in local mode.

def predict(x: LocalDataSet[MiniBatch[T]], batchPerThread: Int)(implicit ev: TensorNumeric[T]): Array[Activity]

Use a model to do prediction in local mode.

def predict(x: RDD[Sample[T]])(implicit ev: TensorNumeric[T]): RDD[Activity]

Use a model to do prediction for RDD.

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

Use a model to do prediction for RDD.

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]], batchPerThread: Int = 4, zeroBasedLabel: Boolean = true): RDD[Int]

Use a model to 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): GraphNet.this.type

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

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

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

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

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

var scaleB: Double

var scaleW: Double

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

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

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

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

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

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

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

implicit val tag: ClassTag[T]

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

def toKeras(): KerasLayer[Activity, Activity, T]

def toString(): String

var train: Boolean

final def training(): GraphNet.this.type

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

def updateGradInput(input: Activity, gradOutput: Activity): Activity

def updateOutput(input: Activity): Activity

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