nebula.core.models.emnist.cnn#
Classes#
Abstract class for the NEBULA model. |
Module Contents#
- class nebula.core.models.emnist.cnn.EMNISTModelCNN(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#
- conv1#
- relu#
- pool1#
- conv2#
- pool2#
- l1#
- l2#
- epoch_global_number#
- forward(x)#
Forward pass of the model.
- configure_optimizers()#
Optimizer configuration.