def map_words(self, _text): mapping = defaultdict(list) tagged_words = pos_tag(set(self.get_words(_text))) for word, tag in tagged_words: mapping[tag].append(word) return mapping