df1dd9b108f3cd7d5130e7f654ea099021c7a2b3,qanta/guesser/experimental/elasticsearch_instance_of.py,ElasticSearchIndex,search,#ElasticSearchIndex#Any#Any#Any#Any#,82

Before Change

                .query(
                    "multi_match",
                    query=text,
                    fields=["wiki_content", "qb_content"]
                )
        else:
            s = Search(index=INDEX_NAME) \
                .query(
                "multi_match",
                query=text,
                fields=["wiki_content", "qb_content"]
            )
        results = s.execute()
        return [(r.page, r.meta.score) for r in results]


es_index = ElasticSearchIndex()

After Change

    class_to_i = {label: i for i, label in enumerate(classes, 1)}
    i_to_class = {i: label for label, i in class_to_i.items()}

    for q, p in zip(questions, pages):
        for sent in q:
            x_data.append(sent)
            if p in instance_of_map:
                y_data.append(class_to_i[instance_of_map[p]])
            else:
                y_data.append(class_to_i[NO_MATCH])

    return x_data, y_data, i_to_class, class_to_i


class ElasticSearchWikidataGuesser(AbstractGuesser):
    def __init__(self):
        super().__init__()
        self.class_to_i = None
        self.i_to_class = None
        self.instance_of_model = None
        guesser_conf = conf["guessers"]["ESWikidata"]
        self.confidence_threshold = guesser_conf["confidence_threshold"]
        self.normalize_score_by_length = guesser_conf["normalize_score_by_length"]

Instances