Project: Named Entity Recognition System

This is the group project for COMP 4211 - Machine Learning. The full name of this project is BERT-based Single-Encoder Multi-Classifier Voting for Named Entity Recognition. My groupmate and I designed a named entity recognition (NER) system based on the BERT model.

In this project, we explored fine tuning the BERT model together with a lot of other machine learning techniques taught in the course COMP 4211.

We used TensorFlow and PyTorch frameworks to build the models. The project is implemented in Python. We also make use of Docker to create an environment so that the graders can run our code without any issues.