How to train on custom dataset

artitw / BERT_QA

Accelerating the development of question-answering systems based on BERT and TF 2.0

https://arxiv.org/abs/1909.05017

Apache License 2.0

19 stars 4 forks source link

Open abhijith-athreya opened 4 years ago

abhijith-athreya commented 4 years ago

Please provide details on how to train on custom QA dataset

artitw commented 4 years ago

One way is to format your custom QA dataset like SQuAD's and then follow the Colab notebook https://colab.research.google.com/drive/1-tLvxSuI0ik2BaruaY_Ivoh_4eobWzEW

Give it a try and let us know how it goes!