ESP32 ML - Model Class Implementation

model.py

class Model:
    def __init__(self, layers, inp_norm_mean=0, inp_norm_sd=1):
        self.layers = layers
	### Subtract the mean divide by SD
        self.inp_norm = lambda inp: normInput(inp, 1/inp_norm_sd, -inp_norm_mean)
    def __call__(self, inp):
        return self.forward(inp)
    def __getitem__(self, i):
        return self.layers[i]
    def forward(self, inp):
        out = self.inp_norm(inp)
        for l in self.layers:
            out = l(out)
        return out