nn.models

fn sequential[T]() &Sequential[T]

sequential creates a new sequential network with a new context.

fn sequential_from_ctx[T](ctx &autograd.Context[T]) &Sequential[T]

sequential_from_ctx creates a new sequential network with the given context.

fn sequential_from_ctx_with_layers[T](ctx &autograd.Context[T], given_layers []types.Layer[T]) &Sequential[T]

sequential_from_ctx_with_layers creates a new sequential network with the given context and the given layers.

fn sequential_info[T](ctx &autograd.Context[T], layers_ []types.Layer[T]) &SequentialInfo[T]

sequential_info creates a new neural network container with an empty list of layers.

fn sequential_with_layers[T](given_layers []types.Layer[T]) &Sequential[T]

sequential_with_layers creates a new sequential network with a new context and the given layers.

fn (mut ls SequentialInfo[T]) input(shape []int)

input adds a new input layer to the network with the given shape.

fn (mut ls SequentialInfo[T]) linear(output_size int)

linear adds a new linear layer to the network with the given output size

fn (mut ls SequentialInfo[T]) maxpool2d(kernel []int, padding []int, stride []int)

maxpool2d adds a new maxpool2d layer to the network with the given kernel size and stride.

fn (mut ls SequentialInfo[T]) mse_loss()

mse_loss sets the loss function to the mean squared error loss.

fn (mut ls SequentialInfo[T]) sigmoid_cross_entropy_loss()

sigmoid_cross_entropy_loss sets the loss function to the sigmoid cross entropy loss.

fn (mut ls SequentialInfo[T]) softmax_cross_entropy_loss()

softmax_cross_entropy_loss sets the loss function to the softmax cross entropy loss.

fn (mut ls SequentialInfo[T]) flatten()

flatten adds a new flatten layer to the network.

fn (mut ls SequentialInfo[T]) relu()

relu adds a new relu layer to the network.

fn (mut ls SequentialInfo[T]) leaky_relu()

leaky_relu adds a new leaky_relu layer to the network.

fn (mut ls SequentialInfo[T]) elu()

elu adds a new elu layer to the network.

fn (mut ls SequentialInfo[T]) sigmoid()

sigmoid adds a new sigmoid layer to the network.

fn (mut ls SequentialInfo[T]) tanh()

tanh adds a new tanh layer to the network.

fn (mut ls SequentialInfo[T]) softmax()

softmax adds a new softmax layer to the network.

fn (mut ls SequentialInfo[T]) gelu()

gelu adds a new GELU layer to the network.

fn (mut ls SequentialInfo[T]) swish()

swish adds a new Swish layer to the network.

fn (mut ls SequentialInfo[T]) mish()

mish adds a new Mish layer to the network.

fn (mut ls SequentialInfo[T]) conv2d(in_channels int, out_channels int, kernel_size []int, config layers.Conv2DConfig)

conv2d adds a new Conv2D layer to the network.

fn (mut ls SequentialInfo[T]) batchnorm1d(num_features int, config layers.BatchNorm1DConfig)

batchnorm1d adds a new BatchNorm1D layer to the network.

fn (mut ls SequentialInfo[T]) avgpool2d(kernel []int, padding []int, stride []int)

avgpool2d adds a new AveragePool2D layer to the network.

fn (mut ls SequentialInfo[T]) global_avgpool2d()

global_avgpool2d adds a new GlobalAveragePool2D layer to the network.

fn (mut ls SequentialInfo[T]) layer_norm(normalized_shape []int, config layers.LayerNormConfig)

layer_norm adds a new LayerNorm layer to the network.

fn (mut ls SequentialInfo[T]) embedding(vocab_size int, embedding_dim int)

embedding adds a new Embedding layer to the network.

fn (mut ls SequentialInfo[T]) lstm(input_size int, hidden_size int, num_layers int)

lstm adds a new LSTM layer to the network.

fn (mut ls SequentialInfo[T]) multihead_attention(embed_dim int, num_heads int)

multihead_attention adds a new MultiHeadAttention layer to the network.

fn (mut ls SequentialInfo[T]) positional_encoding(embed_dim int, max_len int)

positional_encoding adds a new PositionalEncoding layer to the network.

fn (mut ls SequentialInfo[T]) cross_entropy_loss()

cross_entropy_loss sets the loss function to cross entropy loss.

fn (mut ls SequentialInfo[T]) bce_loss()

bce_loss sets the loss function to binary cross entropy loss.

fn (mut ls SequentialInfo[T]) huber_loss()

huber_loss sets the loss function to Huber loss.

fn (mut ls SequentialInfo[T]) nll_loss()

nll_loss sets the loss function to negative log likelihood loss.

fn (mut ls SequentialInfo[T]) kl_div_loss()

kl_div_loss sets the loss function to KL divergence loss.

fn (mut nn Sequential[T]) avgpool2d(kernel []int, padding []int, stride []int)

avgpool2d adds a new AveragePool2D layer to the network.

fn (mut nn Sequential[T]) batchnorm1d(num_features int, config layers.BatchNorm1DConfig)

batchnorm1d adds a new BatchNorm1D layer to the network.

fn (mut nn Sequential[T]) bce_loss()

bce_loss sets the loss function to binary cross entropy loss.

fn (mut nn Sequential[T]) conv2d(in_channels int, out_channels int, kernel_size []int, config layers.Conv2DConfig)

conv2d adds a new Conv2D layer to the network.

fn (mut nn Sequential[T]) cross_entropy_loss()

cross_entropy_loss sets the loss function to cross entropy loss.

fn (mut nn Sequential[T]) elu()

elu adds a new elu layer to the network.

fn (mut nn Sequential[T]) embedding(vocab_size int, embedding_dim int)

embedding adds a new Embedding layer to the network.

fn (mut nn Sequential[T]) flatten()

flatten adds a new flatten layer to the network.

fn (mut nn Sequential[T]) forward(train &autograd.Variable[T]) !&autograd.Variable[T]

fn (mut nn Sequential[T]) gelu()

gelu adds a new GELU layer to the network.

fn (mut nn Sequential[T]) global_avgpool2d()

global_avgpool2d adds a new GlobalAveragePool2D layer to the network.

fn (mut nn Sequential[T]) huber_loss()

huber_loss sets the loss function to Huber loss.

fn (mut nn Sequential[T]) input(shape []int)

input adds a new input layer to the network with the given shape.

fn (mut nn Sequential[T]) kl_div_loss()

kl_div_loss sets the loss function to KL divergence loss.

fn (mut nn Sequential[T]) layer_norm(normalized_shape []int, config layers.LayerNormConfig)

layer_norm adds a new LayerNorm layer to the network.

fn (mut nn Sequential[T]) leaky_relu()

leaky_relu adds a new leaky_relu layer to the network.

fn (mut nn Sequential[T]) linear(output_size int)

linear adds a new linear layer to the network with the given output size

fn (nn &Sequential[T]) load(path string) !

load restores weights into an existing Sequential model from a JSON file. The model's layers must already be constructed in the same order as when saved.

fn (mut nn Sequential[T]) loss(output &autograd.Variable[T], target &vtl.Tensor[T]) !&autograd.Variable[T]

fn (mut nn Sequential[T]) lstm(input_size int, hidden_size int, num_layers int)

lstm adds a new LSTM layer to the network.

fn (mut nn Sequential[T]) maxpool2d(kernel []int, padding []int, stride []int)

maxpool2d adds a new maxpool2d layer to the network with the given kernel size and stride.

fn (mut nn Sequential[T]) mish()

mish adds a new Mish layer to the network.

fn (mut nn Sequential[T]) mse_loss()

mse_loss sets the loss function to the mean squared error loss.

fn (mut nn Sequential[T]) multihead_attention(embed_dim int, num_heads int)

multihead_attention adds a new MultiHeadAttention layer to the network.

fn (mut nn Sequential[T]) nll_loss()

nll_loss sets the loss function to negative log likelihood loss.

fn (mut nn Sequential[T]) positional_encoding(embed_dim int, max_len int)

positional_encoding adds a new PositionalEncoding layer to the network.

fn (mut nn Sequential[T]) relu()

relu adds a new relu layer to the network.

fn (nn &Sequential[T]) save(path string) !

save saves a Sequential model's weights to a JSON file.

fn (mut nn Sequential[T]) set_backend(backend vtl.Backend)

set_backend configures runtime backend preference for all forward passes in this model context.

fn (mut nn Sequential[T]) set_backend_strict(strict bool)

set_backend_strict enables/disables strict runtime backend enforcement.

fn (mut nn Sequential[T]) sigmoid()

sigmoid adds a new sigmoid layer to the network.

fn (mut nn Sequential[T]) sigmoid_cross_entropy_loss()

sigmoid_cross_entropy_loss sets the loss function to the sigmoid cross entropy loss.

fn (mut nn Sequential[T]) softmax()

softmax adds a new softmax layer to the network.

fn (mut nn Sequential[T]) softmax_cross_entropy_loss()

softmax_cross_entropy_loss sets the loss function to the softmax cross entropy loss.

fn (mut nn Sequential[T]) swish()

swish adds a new Swish layer to the network.

fn (mut nn Sequential[T]) tanh()

tanh adds a new tanh layer to the network.

struct Sequential[T] {
pub mut:
	info &SequentialInfo[T] = unsafe { nil }
}

struct SequentialInfo[T] {
pub mut:
	ctx    &autograd.Context[T] = unsafe { nil }
	layers []types.Layer[T]
	loss   types.Loss[T]
}