class

ConstructionKit::Builder

Constructors

new(config : ModelConfig, graph_data : GraphData | Nil = nil, graph_mode : Bool = false)

config

dataset

exec_graph

Graph mode

generate(seed_text : String, max_tokens : Int32 = 100, temperature : Float64 = 0.8) : String

Generate text from a seed

graph_mode?

load_weights(dir : String)

Load all model weights from a directory

model

Legacy mode

save_weights(dir : String)

Save all model weights to a directory

summary

Get model summary info

train_step

Run a single training step, return metrics