def __init__()

in sagemaker_notebook_instance/containers/relationship_extraction/package/models.py [0:0]

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

    def __init__(self, tokenizer, label_encoder, learning_rate=0.0007):
        super().__init__()
        self.model = RelationshipEncoderModule(
            vocab_length=len(tokenizer),
            num_classes=len(label_encoder),
            entity_one_start_token_id=tokenizer.entity_one_start_token_id,
            entity_two_start_token_id=tokenizer.entity_two_start_token_id
        )
        self.learning_rate = learning_rate
        self.train_acc = pl.metrics.Accuracy()
        self.train_f1 = pl.metrics.Fbeta(num_classes=len(label_encoder))
        self.valid_acc = pl.metrics.Accuracy()
        self.valid_f1 = pl.metrics.Fbeta(num_classes=len(label_encoder))