Fine tuning multi[class] classification

Beginners
1

Hello everyone,
I would like to fine tune a LLM model for country code or country name classification.
example: I live in England
answer: GB or Great Britain

Labels will be all county code in the world.

Any advice ? which LLM can I take for this task ?

2

Hi,

An LLM might be a bit of an overkill for this task, you might get good results by fine-tuning a BERT-like text classifier. Also, if a sentence can only belong to one class, the task is actually called multi-class classification instead of multi-label.

See the task guide: Text classification

1 Like
3

This topic was automatically closed 12 hours after the last reply. New replies are no longer allowed.