PhrExt 0.1.0

Extract phrase from English sentence such as noun, verb, preposition based on NLP task word chunking and HuggingFace library

Phrase extraction

Input: PennyLane went to the school

Output: [{'Noun Phrase': 'PennyLane'}, {'Verb Phrase': 'went'}, {'Preposition': 'to'}, {'Noun Phrase': 'the school'}]

I train a sequence tagging model based on RoBERTa of Huggingface. The training code is given in train.py.

How to use

from PhrExt import PhraseExtractor

if __name__ == "__main__":
    phrase_extractor = PhraseExtractor('transZ/phrext', 'roberta-base')
    sent = "PennyLane went to the school"
    res = phrase_extractor(sent)
    print(res)