def build()

in src/diarizers/models/pyannet.py [0:0]

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

    def build(self):
        if self.hparams.linear["num_layers"] > 0:
            in_features = self.hparams.linear["hidden_size"]
        else:
            in_features = self.hparams.lstm["hidden_size"] * (2 if self.hparams.lstm["bidirectional"] else 1)

        self.classifier = nn.Linear(in_features, self.dimension)
        self.activation = self.default_activation()