Edit model card

AmbatronBERTa

AmbatronBERTa is a Thai language model fine-tuned specifically for text classification tasks, built upon the WangchanBERTa architecture.

Model Description

AmbatronBERTa is designed to handle the complexities of the Thai language. It has been fine-tuned on a dataset of over 3,000 research papers to improve classification accuracy. Leveraging the transformer-based WangchanBERTa, it efficiently captures the nuances of Thai text, making it suitable for classifying documents across multiple fields.

Developers

AmbatronBERTa was developed by students at King Mongkut's University of Technology North Bangkok:

  • Peerawat Banpahan
  • Waris Thongpho

Use Cases

AmbatronBERTa can be applied to a wide range of tasks, such as:

  • Research Classification: Categorizing academic papers into relevant topics.
  • Document Organization: Classifying articles, blogs, and other documents by themes.
  • Sentiment Analysis: Analyzing sentiment in Thai-language texts across various contexts.

How to Use

To use AmbatronBERTa with the transformers library:

from transformers import AutoTokenizer, AutoModelForSequenceClassification

# Load the tokenizer and model
tokenizer = AutoTokenizer.from_pretrained("Peerawat2024/AmbatronBERTa")
model = AutoModelForSequenceClassification.from_pretrained("Peerawat2024/AmbatronBERTa")
Downloads last month
530
Safetensors
Model size
105M params
Tensor type
F32
·
Inference API
Unable to determine this model's library. Check the docs .

Model tree for Peerawat2024/AmbatronBERTa

Finetuned
(30)
this model