def forward()

in code/model_def.py [0:0]

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

    def forward(self, input_ids, attention_mask):
        output = self.bert(
          input_ids=input_ids,
          attention_mask=attention_mask
        )
        return self.classifier(output.pooler_output)