Home

Awesome

Thai Intent Classification Using BERT

This repository contains python code to Finetune BERT for Thai intent classification with GPU on Google Colab using the pretrained model from https://github.com/ThAIKeras/bert which uses data from Thai Wikipedia. This can be applied to create an advance chatbot which can be integrated with robots, websites, apps and etc.

This project is pretty much similar to https://github.com/KongpolC/Thai-Sentiment-Classification-Using-BERT so, feel free to check this one out too.

Data

Trained on truevoice-intent dataset which was provided by TrueVoice. This dataset contains text transcriptions from customer service phone calls and there destination intents (TrueVoice's Mari). Source: https://github.com/PyThaiNLP/truevoice-intent

Transfer Learning Using GPU

Main processes to fintune BERT in Thai using GPU on Google Colab include:

Result

2 epochs: accuracy = 0.8921

confusion_matrix