8a2e857746dbab697b0045aa6c38133f3de71250,nalaf/learning/taggers.py,StubSameSentenceRelationExtractor,annotate,#StubSameSentenceRelationExtractor#Any#,138

Before Change

        self.sentence_splitter.split(dataset)
        self.tokenizer.tokenize(dataset)

        for document in dataset:
            for part in document:
                for ann_1, ann_2 in product(
                    (a for a in part.annotations if a.class_id == self.entity1_class),
                    (a for a in part.annotations if a.class_id == self.entity2_class)):

                    if part.get_sentence_index_for_annotation(ann_1) == part.get_sentence_index_for_annotation(ann_2):
                        rel = Relation(self.relation_type, ann_1, ann_2)
                        part.predicted_relations.append(rel)


class StubSameDocumentPartRelationExtractor(RelationExtractor):
    // Would be better if it just ussed an existing edge generator

    def __init__(self, entity1_class, entity2_class, relation_type):

After Change

    def annotate(self, dataset):
        super().annotate(dataset)


class StubSameDocumentPartRelationExtractor(RelationExtractor):

Instances