Skip to content

Latest commit

 

History

History
10 lines (7 loc) · 573 Bytes

README.md

File metadata and controls

10 lines (7 loc) · 573 Bytes

BERT-Fine-Tuning-on-Toxic-Comment-Regression

Final project for Statistical Language Modeling class at Northwestern University.

Only main BERT files have been included in this repo. In order to run the code, a separate BERT directory with initial model weights should be added. Additionally, when running, the model needs to have an output directory defined. From this output directory, inference can be run after fine-tuning.

Image_1 Image_2 Image_3 Image_4