Skip to content

soroushjavdan/NLIbert

Repository files navigation

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

Releases

No releases published

Packages

No packages published

Languages