o net cringe

def baza(n_layers: list, acts: object) -> list:

    models = []

    for i in n_layers:

        for act in acts:

            layers = []

            in_features = data_post[0].shape[1]

            for j in range(i):

                out_features = math.floor(in_features/2)

                layers.append(nn.Linear(in_features, out_features))
                layers.append(act)

                in_features = out_features

            layers.append(nn.Linear(in_features, 1))
            layers.append(act)

            models.append(nn.Sequential(*layers))

    return models

hidden_layers = [1,2,3,4]
acts = [nn.LeakyReLU(), nn.Tanh(),nn.Sigmoid()]
models = baza(hidden_layers, acts)