This repository contains code for the Question Answering part of our project work for NLP2020 class by Paolo Torroni @unibo.
It implements 2 main architectures: a BiLSTM fully trained model and one based on BERT fine-tuning.
This folder contains:
compute_answer.py
: given the question file, this script will download the best model and save the predictionsQA.ipynb
: the notebook used for train and evaluate the best model
- rnn: https://wandb.ai/veri/SQUAD/reports/RNN--Vmlldzo1Mzk2NTU
- transformers: https://wandb.ai/buio/SQUAD/reports/Transformers-Report--Vmlldzo1Mzk3MjE
main
: merged from thehuggingface
branchrnn
: baseline model based on RNNrnn-regression
: experiment with RNN + regression headshuggingface
: transformer-base models, comparation between BERT, ELECTRA, RoBERTa, Longformerhuggingface-regression
: experiment with RoBERTa + regression head