raajanwankhade/BERT-Dialogue_Classification

A dialogue classifier from the show The Office. This code utilises the Bert (Base) model to classify the dialogues as Jim or Dwight's Dialogues.

Jupyter Notebook