Functional

fun Functional(vararg layers: Layer)

Creates a Functional model via sequence of layers.

Companion

object Companion

close

open override fun close()

Closes internal resources: session and kGraph.

compile

open override fun compile(optimizer: Optimizer, loss: LossFunction, metric: Metric, callback: Callback)

open override fun compile(optimizer: Optimizer, loss: LossFunction, metric: Metrics, callback: Callback)

open override fun compile(optimizer: Optimizer, loss: Losses, metric: Metric, callback: Callback)

open override fun compile(optimizer: Optimizer, loss: Losses, metric: Metrics, callback: Callback)

Configures the model for training.

copy

fun copy(saveOptimizerState: Boolean = false, copyWeights: Boolean = true): Functional

Returns a copy of this model.

open override fun copy(copiedModelName: String?, saveOptimizerState: Boolean, copyWeights: Boolean): TensorFlowInferenceModel

Creates a copy.

evaluate

open override fun evaluate(dataset: Dataset, batchSize: Int): EvaluationResult

Returns the metrics and loss values for the model in test (evaluation) mode.

fun evaluate(dataset: Dataset, metric: Metrics): Double

Evaluates dataset via metric.

fit

open override fun fit(dataset: Dataset, epochs: Int, batchSize: Int): TrainingHistory

open override fun fit(trainingDataset: Dataset, validationDataset: Dataset, epochs: Int, trainBatchSize: Int, validationBatchSize: Int): TrainingHistory

Trains the model for a fixed number of epochs (iterations over a dataset).

fun fit(dataset: OnHeapDataset, validationRate: Double, epochs: Int, trainBatchSize: Int, validationBatchSize: Int): TrainingHistory

Trains the model for a fixed number of epochs (iterations on a dataset).

getLayer

infix fun getLayer(layerName: String): Layer

Return layer by layerName.

graphToString

fun graphToString(): String

Forms the graph description in string format.

init

fun init()

Initializes kGraph variables.

input

fun input(inputOp: Input)

Chain-like setter to set up inputOp.

kGraph

fun kGraph(): KGraph

Returns KGraph.

loadWeights

open override fun loadWeights(modelDirectory: File, loadOptimizerState: Boolean)

Loads variable data from .txt files.

output

fun output(outputOp: Output)

Chain-like setter to set up outputOp.

predict

open override fun predict(inputData: FloatArray): Int

Generates output prediction for the input sample.

fun predict(dataset: Dataset): List<Int>

Predicts labels for all observation in dataset.

open override fun predict(inputData: FloatArray, predictionTensorName: String): Int

Generates output prediction for the input sample using output of the predictionTensorName tensor.

open override fun predict(dataset: Dataset, batchSize: Int): IntArray

Generates output predictions for the input samples.

predictAndGetActivations

open override fun predictAndGetActivations(inputData: FloatArray, predictionTensorName: String): Pair<Int, List<*>>

Predicts and returns not only prediction but list of activations values from intermediate model layers (for visualisation or debugging purposes).

predictSoftly

open override fun predictSoftly(inputData: FloatArray, predictionTensorName: String): FloatArray

Predicts vector of probabilities instead of specific class in predict method.

open override fun predictSoftly(dataset: Dataset, batchSize: Int): Array<FloatArray>

Generates output predictions for the input samples. Each prediction is a vector of probabilities instead of specific class in predict method.

reshape

open override fun reshape(vararg dims: Long)

Chain-like setter to set up input shape.

save

open override fun save(modelDirectory: File, savingFormat: SavingFormat, saveOptimizerState: Boolean, writingMode: WritingMode)

Saves the model as graph and weights.

summary

open override fun summary(): ModelSummary

Returns model summary.

toString

open override fun toString(): String

inputDimensions

open override val inputDimensions: LongArray

Returns input dimensions in order HWC (height, width, channels)

inputLayer

val inputLayer: Input

First layer that is responsible for the input shape of the Neural Network.

isBuiltForForwardMode

var isBuiltForForwardMode: Boolean = false

Is true when model is ready for forward mode.

isModelCompiled

var isModelCompiled: Boolean = false

Is true when model is compiled.

isModelInitialized

var isModelInitialized: Boolean = false

Is true when model is initialized.

isOptimizerVariableInitialized

var isOptimizerVariableInitialized: Boolean = false

Is true when model optimizer variables are initialized.

kGraph

lateinit var kGraph: KGraph

TensorFlow wrapped computational graph.

layers

var layers: List<Layer>

The layers to describe the model design. Main part of the internal state of the model.

logger

val logger: KLogger

Logger for the model.

loss

var loss: LossFunction

Loss function.

name

var name: String? = null

Model name.

numberOfClasses

var numberOfClasses: Long

Number of classes for classification tasks. -1 is a default value for regression tasks.

shape

lateinit var shape: LongArray

Data shape for prediction.

stopTraining

var stopTraining: Boolean = false

Special flag for callbacks.

loadWeightsForFrozenLayersByPathTemplates

fun Functional.loadWeightsForFrozenLayersByPathTemplates(hdfFile: HdfFile, kernelDataPathTemplate: String = KERNEL_DATA_PATH_TEMPLATE, biasDataPathTemplate: String = BIAS_DATA_PATH_TEMPLATE)

Loads weights from hdf5 file created in Keras TensorFlow framework for non-trainable (or frozen) layers only.

Functional

Constructors

Types

Functions

Properties

Extensions