Jaewoo Song
Jaewoo Song

Categories

  • Tech

To evaluate various models based on BERT structure on dialogue understanding tasks, I created my own project to do it.

This repository can test the capability of a model on several tasks related to Task-oriented dialogues, including Intent Detection, Entity Recognition and Action Prediction.

It contains the codes for pre-processing each dataset corresponding to each task, loading different encoder model, fine-tuning it and testing after training.

So in this post, I introduce the details of this package and explain several parts of the entire codes.





devjwsong/dialogue-understanding-tasks-pytorch. https://github.com/devjwsong/dialogue-understanding-tasks-pytorch.