GraphTrainableModelBuilder

class GraphTrainableModelBuilder<T : GraphTrainableModel>(createModelFunction: (Array<Layer>) -> T)

Graph trainable model builder.

Parameters

T

Type of model.

Constructors

GraphTrainableModelBuilder

fun <T : GraphTrainableModel> GraphTrainableModelBuilder(createModelFunction: (Array<Layer>) -> T)

Creates Graph trainable model builder.

Functions

layers

fun layers(block: LayerListBuilder.() -> Unit)

Layers Builder.

model

fun model(block: T.() -> Unit)

Model Builder.

use

fun use(block: T.() -> Unit)

Use builder.