def get_verbs(text, lowercase=False, lemmatize=False) -> Generator: """ Get only verbs from text using Stanford libraries. :param text: :param lowercase: :param lemmatize: :return: """ if not is_stanford_enabled(): raise RuntimeError( "USE_STANFORD is set to False. No Stanford functionality available." ) # Get tokens and tag tokens = get_tokens_list(text) pos = STANFORD_TAGGER.tag(tokens) verb_index = [i for i in range(len(pos)) if pos[i][1].startswith("V")] if lemmatize: lemmas = get_lemma_list(text, lowercase=lowercase) for j in verb_index: yield lemmas[j] else: for j in verb_index: yield tokens[j].lower() if lowercase else tokens[j]
def get_verbs(text, lowercase=False, lemmatize=False) -> Generator: """ Get only verbs from text using Stanford libraries. :param text: :param lowercase: :param lemmatize: :return: """ check_stanford() # Get tokens and tag tokens = get_tokens_list(text) pos = STANFORD_TAGGER.tag(tokens) verb_index = [i for i in range(len(pos)) if pos[i][1].startswith("V")] if lemmatize: lemmas = get_lemma_list(text, lowercase=lowercase) for j in verb_index: yield lemmas[j] else: for j in verb_index: yield tokens[j].lower() if lowercase else tokens[j]