Views
No views yet
t5-small pretrained model, fine-tuned on the task of generating QANom QAs.1import transformers
2model = transformers.AutoModelForSeq2SeqLM.from_pretrained("kleinay/qanom-seq2seq-model-baseline")
3tokenizer = transformers.AutoTokenizer.from_pretrained("kleinay/qanom-seq2seq-model-baseline")pipeline.py file from this repository, and then use the QASRL_Pipeline class:1from pipeline import QASRL_Pipeline
2pipe = QASRL_Pipeline("kleinay/qanom-seq2seq-model-baseline")
3pipe("The student was interested in Luke 's <predicate> research about see animals .", verb_form="research", predicate_type="nominal")1[{'generated_text': 'who _ _ researched something _ _ ?<extra_id_7> Luke',
2 'QAs': [{'question': 'who researched something ?', 'answers': ['Luke']}]}]transformers.pipelines in the official docs.<predicate> symbol, but you can also specify your own predicate marker:pipe("The student was interested in Luke 's <PRED> research about see animals .", verb_form="research", predicate_type="nominal", predicate_marker="<PRED>")pipe("The student was interested in Luke 's <predicate> research about see animals .", verb_form="research", predicate_type="nominal", num_beams=3)