You need to agree to share your contact information to access this model

This repository is publicly accessible, but you have to accept the conditions to access its files and content.

Log in or Sign Up to review the conditions and access this model content.

QomSSLab/SubjectClassifier-v1

This repository hosts a sequence-classification head trained for text classification.

Usage

from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline

model_id = "QomSSLab/SubjectClassifier-v1"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForSequenceClassification.from_pretrained(model_id)
classifier = pipeline("text-classification", model=model, tokenizer=tokenizer)

text = "مثال از یک ورودی فارسی"
print(classifier(text))

Labels

  • اخلاق
  • تاریخ
  • حدیث
  • سیاسی
  • فقه و احکام
  • قرآن
  • مشاوره
  • کلام و اعتقادات

Metrics

Validation Metrics

  • Precision: 0.9717
  • Recall: 0.9713
  • F1: 0.9713
  • Accuracy: 0.9713

Per-label Breakdown

Label Precision Recall F1 Support
اخلاق 0.9455 0.9598 0.9526 199
تاریخ 0.9815 1.0000 0.9907 53
حدیث 0.9487 0.9823 0.9652 113
سیاسی 1.0000 0.9693 0.9844 163
فقه و احکام 0.9652 1.0000 0.9823 222
قرآن 1.0000 0.9875 0.9937 80
مشاوره 0.9670 0.9263 0.9462 95
کلام و اعتقادات 0.9818 0.9558 0.9686 226
Downloads last month
-
Safetensors
Model size
0.6B params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support