b858e124f53677a0956257b4b7077c407c3d4d88,solutions/set_expansion/set_expand.py,SetExpand,in_vocab,#SetExpand#Any#,99

Before Change

        return [self.__id2term(id) for id in self.np2vec_model.vocab]

    def in_vocab(self, term):
        norm = None
        logger.info("check is in vocab: " + term)
        if term in self.np2id.keys():
            norm = self.np2id[term]
        if norm is None or norm not in self.id2rep or self.__term2id(
                    self.id2rep[norm]) not in self.np2vec_model.vocab:
            return False
        return True

    def get_group(self, term):
        logger.info("check is in vocab: " + term)

After Change

        return [self.__id2term(id) for id in self.np2vec_model.vocab]

    def in_vocab(self, term):
        id = self.__term2id(term)
        if id is None:
            return False
        return True

    def get_group(self, term):
        logger.info("get group of: " + term)

Instances