Neural Network

The NeuralNetwork class is the base model for a feed forward neural network.

Parameters:

  • layers: A list of Layer objects.

  • loss_fn (Optional) : A LossFunction object. By default uses Mean Squared Error.

Returns:

class NeuralNetwork:
    def __init__(self, layers: Optional[list[Layer]] = [], loss_fn: Optional[LossFunction] = MeanSquaredError):
        self.layers = layers
        self.compiled = False
        self.loss_fn = loss_fn

Last updated