Vietnam Tourism Named Entity Recognition

We fine-tuned BERT to train Vietnam tourism dataset for a question answering system. The model was called NER2QUES because it detected tourism NER in a sentence. From that, the system generated questions corresponding to NER types.

How to use

You can use in Transformers

from transformers import AutoTokenizer, AutoModelForTokenClassification

tokenizer = AutoTokenizer.from_pretrained("truongphan/vntourismNER")

model = AutoModelForTokenClassification.from_pretrained("truongphan/vntourismNER")

line = "King Garden is located in Thanh Thuy, Phu Tho"
l.append(line)

predictions, raw_outputs = model.predict(l)
print(predictions)

Authors

Phuc Do, University of Information Technology, Ho Chi Minh national university, Vietnam
Truong H. V. Phan, Van Lang university, Ho Chi Minh city, Vietnam