def __init__()

in svhn_based_exp/models.py [0:0]

• 5 lines of code
• 2 McCabe index (conditional complexity)

    def __init__(self, in_features, num_classes, hidden_layers=[]):
        super(MLP, self).__init__()
        width = [in_features] + hidden_layers + [num_classes]
        self.layers = nn.ModuleList([nn.Linear(width[i], width[i + 1]) for i in range(len(width) - 2)])
        self.output_layer = nn.Linear(width[-2], width[-1])