subject-clf-distilbert
Multi-label subject classifier -- labels: English, Maths, Science, Civics, Geology.
- base:
distilbert-base-uncased - trained on
yanlanh/multilabel-subject-dataset@5d8dc42
from transformers import AutoModelForSequenceClassification, AutoTokenizer
tok = AutoTokenizer.from_pretrained("yanlanh/subject-clf-distilbert")
model = AutoModelForSequenceClassification.from_pretrained("yanlanh/subject-clf-distilbert")
- Downloads last month
- 3