Skip to content Skip to sidebar Skip to footer

Fine-tune Bert For Specific Domain (unsupervised)

I want to fine-tune BERT on texts that are related to a specific domain (in my case related to engineering). The training should be unsupervised since I don't have any labels or an

Solution 1:

What you in fact want to is continue pre-training BERT on text from your specific domain. What you do in this case is to continue training the model as masked language model, but on your domain-specific data.

You can use the run_mlm.py script from the Huggingface's Transformers.

Post a Comment for "Fine-tune Bert For Specific Domain (unsupervised)"