Skip to content

Intent classification using HerBERT language model and a custom intent classification head.

Notifications You must be signed in to change notification settings

rz4b4/Intent-classification-Polish-language

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

20 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Intent-classification-Polish-language

A deep learning solution for intent classification problem in the domain of Polish language.

Approach

Intent classification is done by using a HerBERT as a backbone language model to understand the utterances. On top of that, a custom intent classification head is used to output the utterance intent.

Dataset

MASSIVE: https://arxiv.org/pdf/2204.08582.pdf

Useful files

  • train.py - train code
  • test.py - test code with csv export for further usage in the analytics.py
  • analytics.py - code that extracts interesting information about the trained model, it uses the csv file exported using test.py code
  • models/best_model - this folder contains the config and model definition for a training setting, that achieved ~86% validation accuracy

About

Intent classification using HerBERT language model and a custom intent classification head.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages