def __init__()

in review_sentiment_model.py [0:0]

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

    def __init__(self, transformer, output_dim, freeze):
        super().__init__()
        self.transformer = transformer
        hidden_dim = transformer.config.hidden_size
        self.fc = nn.Linear(hidden_dim, output_dim)
        if freeze:
            for param in self.transformer.parameters():
                param.requires_grad = False