def __init__()

in ai-judge/model.py [0:0]

• 13 lines of code
• 1 McCabe index (conditional complexity)

    def __init__(self):
        super(CNN, self).__init__()
        
        self.conv2D_1 = Sequential(
            Conv2d(1, 16, kernel_size=5, stride=2, padding=2),
            ReLU(),
            MaxPool2d(kernel_size=2, stride=2))
        self.conv2D_2 = Sequential(
            Conv2d(16, 32, kernel_size=5, stride=2, padding=2),
            ReLU(),
            AdaptiveMaxPool2d((5, 5)))
        self.dropout = Dropout(0.5)
        self.linear_1 = Linear(5 * 5 * 32, 100)
        self.linear_2 = Linear(100, 2)