ebe4256914107273feb1da940fc3e4237731365e,nalaf/preprocessing/edges.py,SimpleEdgeGenerator,generate,#SimpleEdgeGenerator#,37

Before Change

    def generate(self, dataset):
        from itertools import product, chain

        for part in dataset.parts():
            part.edges = []

            for ann_1, ann_2 in product(
                    (ann for ann in chain(part.annotations, part.predicted_annotations) if ann.class_id == self.entity1_class),
                    (ann for ann in chain(part.annotations, part.predicted_annotations) if ann.class_id == self.entity2_class)):

                index_1 = part.get_sentence_index_for_annotation(ann_1)
                index_2 = part.get_sentence_index_for_annotation(ann_2)

                if index_1 == index_2 and index_1 is not None:
                    part.edges.append(
                        Edge(ann_1, ann_2, self.relation_type, index_1, part))


class WordFilterEdgeGenerator(EdgeGenerator):
    
    Simple implementation of generating edges between the two entities
    if they are contained in the same sentence AND the sentence

After Change

    def generate(self, dataset):
        import warnings
        warnings.warn("Use `SentenceDistanceEdgeGenerator` directly. This will be deleted", DeprecationWarning)

        super().generate(dataset)

Instances