def __init__()

in fast_grad_models.py [0:0]

• 10 lines of code
• 3 McCabe index (conditional complexity)

    def __init__(self, hidden_sizes):
        super(type(self), self).__init__()

        assert(len(hidden_sizes) >= 2)
        self.input_size = hidden_sizes[0]
        self.act = F.relu

        if len(hidden_sizes) == 2:
            self.hidden_layers = []
        else:
            self.hidden_layers = nn.ModuleList([nn.Linear(hidden_sizes[i], hidden_sizes[i+1]) for i in range(len(hidden_sizes) - 2)])
        self.output_layer = nn.Linear(hidden_sizes[-2], hidden_sizes[-1])