TrainableModel

fun TrainableModel()

close

open override fun close()

Closes internal resources: session and kGraph.

compile

abstract fun compile(optimizer: Optimizer, loss: LossFunction, metric: Metric, callback: Callback = Callback())

abstract fun compile(optimizer: Optimizer, loss: LossFunction, metric: Metrics, callback: Callback = Callback())

abstract fun compile(optimizer: Optimizer, loss: Losses, metric: Metric, callback: Callback = Callback())

abstract fun compile(optimizer: Optimizer, loss: Losses, metric: Metrics, callback: Callback = Callback())

Configures the model for training.

copy

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

Creates a copy.

evaluate

abstract fun evaluate(dataset: Dataset, batchSize: Int = 256): 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

abstract fun fit(dataset: Dataset, epochs: Int = 5, batchSize: Int = 32): TrainingHistory

abstract fun fit(trainingDataset: Dataset, validationDataset: Dataset, epochs: Int = 5, trainBatchSize: Int = 32, validationBatchSize: Int = 256): 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).

graphToString

fun graphToString(): String

Forms the graph description in string format.

input

fun input(inputOp: Input)

Chain-like setter to set up inputOp.

loadWeights

open fun loadWeights(modelDirectory: File, loadOptimizerState: Boolean = false)

Loads variable data from .txt files.

output

fun output(outputOp: Output)

Chain-like setter to set up outputOp.

predict

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

Predicts labels for all observation in dataset.

open override fun predict(inputData: FloatArray): Int

Generates output prediction for the input sample.

abstract fun predict(inputData: FloatArray, predictionTensorName: String): Int

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

abstract fun predict(dataset: Dataset, batchSize: Int): IntArray

Generates output predictions for the input samples.

predictAndGetActivations

abstract 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

abstract 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.

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

Predicts 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

abstract fun save(modelDirectory: File, savingFormat: SavingFormat = SavingFormat.TF_GRAPH_CUSTOM_VARIABLES, saveOptimizerState: Boolean = false, writingMode: WritingMode = WritingMode.FAIL_IF_EXISTS)

Saves the model as graph and weights.

summary

abstract fun summary(): ModelSummary

Returns model summary.

toString

open override fun toString(): String

inputDimensions

open override val inputDimensions: LongArray

Input specification for this model.

isBuiltForForwardMode

var isBuiltForForwardMode: Boolean = false

Is true when model is ready for forward mode.