Neural Network
The NeuralNetwork class is the base model for a feed forward neural network.
Parameters:
layers
: A list ofLayer
objects.loss_fn
(Optional) : ALossFunction
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