Downloads · 30 days
8
22% of all-time downloads
hw2_text_finetune_distilbert
george2cool36/hw2_text_finetune_distilbert
hw2_text_finetune_distilbert is a text classification model from george2cool36. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as mit.
Fine-tuned DistilBERT for text classification on a classmate's HW1 dataset.
All-time downloads
37
Public
Parameters
67M
4 GB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pt536 MB · 50%
At a glance
- Task
- Text Classification
- Library
- transformers
- License
- mit
- Model type
- distilbert
- Access
- Public
- Created
- Sep 22, 2025
- Updated
- Sep 23, 2025
- SHA
- 18a90f49
Try a prompt
Datasets
Model card
From the Hugging Face model README
HW 2 - DistilBERT fine-tuned
Task
Fine-tuned DistilBERT for text classification on a classmate's HW1 dataset.
- Dataset:
ddecosmo/hw_text_dataset - Text column:
Text - Label column:
label(classes: ['asu', 'bucknell', 'cmu', 'duq', 'ucsd']) - Train/Eval split: 80/20 (stratified if available)
Training
- Base model:
distilbert-base-uncased - Epochs: 3, LR=5e-5, WD=0.01, warmup=10%
- Batch size: 16
- Best model by: F1 (macro)
Results (Test)
- Accuracy: 0.4000
- F1 (macro): 0.1231
- Precision (macro): nan
- Recall (macro): nan
Notes & Limitations
- Small student dataset; results may vary with seeds.
- Labels mapped as: {'asu': 0, 'bucknell': 1, 'cmu': 2, 'duq': 3, 'ucsd': 4}
AI Tool Disclosure
This notebook used ChatGPT for scaffolding code and documentation. All dataset selection, training, evaluation, and uploads were performed by the student.
- Task
- Text Classification
- Library
- transformers
- Type
- distilbert
- License
- mit
- Created
- Sep 22, 2025
- Updated
- Sep 23, 2025