nebula.core.models.kitsun.mlp

Classes

KitsunModelMLP

Abstract class for the NEBULA model.

Module Contents

class nebula.core.models.kitsun.mlp.KitsunModelMLP(input_channels=3, num_classes=10, learning_rate=0.001, metrics=None, confusion_matrix=None, seed=None)

Bases: nebula.core.models.nebulamodel.NebulaModel

Abstract class for the NEBULA model.

This class is an abstract class that defines the interface for the NEBULA model.

config

example_input_array

learning_rate

criterion

l1

batchnorm1

dropout

l2

batchnorm2

l3

batchnorm3

l5

forward(x)

Forward pass of the model.

configure_optimizers()

Optimizer configuration.