This repository contains Jupyter notebooks for fine-tuning BERT models for specific NLP tasks:
- Medical Transcripts Question Answering (MTQA)
- Cybersecurity Named Entity Recognition (CyberNER)
The notebooks provide step-by-step guidance on loading pre-trained BERT models, preparing datasets, and fine-tuning for these specific tasks.
MTQA_FineTuning.ipynb
: Fine-tunes a BERT model for Medical Transcripts QA and demonstrates how to use the fine-tuned model for inference.
CyberNER_FineTuning.ipynb
: Fine-tunes a BERT model for Cybersecurity NER and demonstrates how to use the fine-tuned model for inference.
Click on Open in Colab
within the notebook in GitHub to get started.
The data directory contains datasets for MTQA and CyberNER.
If you encounter any issues or have suggestions, please open a GitHub issue. Your feedback is valuable.