agentlans / GIST-small-en-domain-classifier

GIST-small-en-domain-classifier

A fine-tuned version of the bert architecture ( BertForSequenceClassification ) optimized for the text-classification task.

• Model type: bert
• Problem Type: single_label_classification
• Number of Labels: 26
• Vocabulary Size: 30522
• License: MIT

Use

To get started with this model in Python using the Hugging Face Transformers library, run the following code:

from transformers import AutoTokenizer, AutoModelForSequenceClassification
import torch

model_id = "agentlans/GIST-small-en-domain-classifier"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForSequenceClassification.from_pretrained(model_id)

text = "Replace this with your input text."
inputs = tokenizer(text, return_tensors="pt")

with torch.no_grad():
    logits = model(**inputs).logits

predicted_class_id = logits.argmax().item()
predicted_class_name = model.config.id2label[predicted_class_id]

print(f"Predicted Class ID: {predicted_class_id}")
print(f"Predicted Class Name: {predicted_class_name}")

Intended Uses & Limitations

Intended Use

This model is designed for sequence classification tasks. Below are the specific class labels mapped to their corresponding IDs:

Training Details

Hyperparameters

The following hyperparameters were used during fine-tuning:

• Learning Rate: 5e-05
• Train Batch Size: 8
• Eval Batch Size: 8
• Optimizer: OptimizerNames.ADAMW_TORCH_FUSED
• Number of Epochs: 3.0
• Mixed Precision: BF16

Training Results & Evaluation

During fine-tuning, the model achieved the following results on the evaluation set:

Speed Performance

• Training Runtime: 5171.9832 seconds
• Train Samples per Second: 464.039
• Evaluation Runtime: 56.3341 seconds
• Eval Samples per Second: 1775.159

Framework Versions

• Transformers: 5.0.0.dev0
• PyTorch: 2.9.1+cu128