Package org.jetbrains.kotlinx.dl.api.core

Types

Functional

class Functional(layers: Layer) : GraphTrainableModel

A Functional model is defined as a directed graph of layers.

GraphTrainableModel

abstract class GraphTrainableModel(layers: Layer) : TrainableModel

GraphTrainableModel model groups a linear stack of layers into a graph-based TensorFlow Model. Also, it provides training and inference features on this model.

KGraph

class KGraph(graphDef: ByteArray, prefix: String) : AutoCloseable

A wrapper object to extend functionality and behaviour of static TensorFlow graph.

SavingFormat

enum SavingFormat : Enum<SavingFormat>

Model saving format.

Sequential

class Sequential(layers: Layer) : GraphTrainableModel

Sequential model groups a linear stack of layers into a TensorFlow Model. Also, it provides training and inference features on this model.

TrainableModel

abstract class TrainableModel : TensorFlowInferenceModel

Base abstract class for all trainable models.

WritingMode

enum WritingMode : Enum<WritingMode>

Model writing mode.

Functions

freeze

fun GraphTrainableModel.freeze()

Freezes weights in all layers in this model, so they won't be changed during training.