NLIbert
NLIbert copied to clipboard
Dialogue Natural Language Inference with bert classifier
trafficstars
NLIbert
Dialogue Natural Language Inference with bert classifier.
Description
In this code we make use of dataset called Dialogue NLI.
How to:
First
you need to download the DataSet from the below link.
https://wellecks.github.io/dialogue_nli/
Then place the downloaded data inside data/dialogue_nli_extra directory.
Second
because of problem during compression process the training file need to be repair. run the below command in order to repair it.
❱❱❱ python data_repairer.py
Third
install the requirements
❱❱❱ pip install -r requirements.txt
Now we are good to go !!
Training phase
just run the below command
❱❱❱ python train.py --gpu --data_path data/dialogue_nli_extra/ --save_path save/ --lr 5e-5 --batch_size 32 --epochs 4 --plot_path save/plot/ --bert_model bert-base-cased