Method readable_label

python/graphvite/dataset.py:910–942

(self, labels, save_file, hierarchy=False)

        return name[:name.find(".")]

    def readable_label(self, labels, save_file, hierarchy=False):
        wordnet = self.import_wordnet()

        if hierarchy:
            logger.info("generating human-readable hierarchical labels")
        else:
            logger.info("generating human-readable labels")
        synsets = []
        for label in labels:
            pos = label[0]
            offset = int(label[1:])
            synset = wordnet.synset_from_pos_and_offset(pos, offset)
            synsets.append(synset)
        depth = max([synset.max_depth() for synset in synsets])

        num_sample = len(synsets)
        labels = [self.get_name(synset) for synset in synsets]
        num_class = len(set(labels))
        hierarchies = [labels]
        while hierarchy and num_class > 1:
            depth -= 1
            for i in range(num_sample):
                if synsets[i].max_depth() > depth:
                    # only takes the first recall
                    synsets[i] = synsets[i].hypernyms()[0]
            labels = [self.get_name(synset) for synset in synsets]
            hierarchies.append(labels)
            num_class = len(set(labels))
        hierarchies = hierarchies[::-1]

        with open(save_file, "w") as fout:
            for hierarchy in zip(*hierarchies):
                fout.write("%s\n" % "\t".join(hierarchy))

    def image_feature_data(self, image_path):
        """"""