def predict(self, x):
    for layer in self.layers.values():
        x = layer.forward(x)
    return x
def predict(self, x):
    for layer in self.layers.values():
        x = layer.forward(x)
    return x