Add canonical model card — v6 Kaggle T4, 93.95% accuracy
Browse files
README.md
CHANGED
|
@@ -4,9 +4,9 @@ language:
|
|
| 4 |
license: apache-2.0
|
| 5 |
tags:
|
| 6 |
- text-classification
|
| 7 |
-
- emotion-classification
|
| 8 |
-
- roberta
|
| 9 |
- the-founder
|
|
|
|
|
|
|
| 10 |
datasets:
|
| 11 |
- dair-ai/emotion
|
| 12 |
metrics:
|
|
@@ -18,9 +18,9 @@ model-index:
|
|
| 18 |
results:
|
| 19 |
- task:
|
| 20 |
type: text-classification
|
| 21 |
-
name:
|
| 22 |
dataset:
|
| 23 |
-
name:
|
| 24 |
type: dair-ai/emotion
|
| 25 |
split: validation
|
| 26 |
metrics:
|
|
@@ -28,7 +28,7 @@ model-index:
|
|
| 28 |
value: 0.9395
|
| 29 |
name: Accuracy
|
| 30 |
- type: loss
|
| 31 |
-
value: 0.
|
| 32 |
name: Eval Loss
|
| 33 |
---
|
| 34 |
|
|
@@ -38,57 +38,69 @@ Fine-tuned by **The Founder** — an autonomous ML orchestration superagent.
|
|
| 38 |
|
| 39 |
## Model Description
|
| 40 |
|
| 41 |
-
|
|
|
|
|
|
|
| 42 |
|
| 43 |
## Model Details
|
| 44 |
|
| 45 |
| Property | Value |
|
| 46 |
|---|---|
|
| 47 |
| Base model | roberta-base |
|
| 48 |
-
|
|
|
|
|
| 49 |
| Epochs | 4 |
|
| 50 |
| Batch size | 32 |
|
| 51 |
| Learning rate | 2e-05 |
|
| 52 |
| GPU | Tesla T4 (Kaggle) |
|
| 53 |
-
| Train loss | 0.
|
| 54 |
-
| Eval loss | 0.
|
| 55 |
| Eval accuracy | 0.9395 |
|
| 56 |
-
| Duration |
|
| 57 |
|
| 58 |
## How to Get Started
|
| 59 |
|
| 60 |
```python
|
| 61 |
from transformers import pipeline
|
| 62 |
clf = pipeline("text-classification", model="zanesmit29/founder-roberta-emotion-v1")
|
| 63 |
-
clf("
|
| 64 |
```
|
| 65 |
|
| 66 |
## Uses
|
| 67 |
|
| 68 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 69 |
|
| 70 |
## Training Details
|
| 71 |
|
| 72 |
### Data
|
| 73 |
-
Trained on `dair-ai/emotion` — 4 epochs.
|
| 74 |
|
| 75 |
### Hyperparameters
|
| 76 |
|
| 77 |
-
|
|
| 78 |
|---|---|
|
| 79 |
-
|
|
| 80 |
-
| Batch | 32 |
|
| 81 |
| Epochs | 4 |
|
|
|
|
|
|
|
|
|
|
| 82 |
| fp16 | true |
|
| 83 |
|
| 84 |
## Results
|
| 85 |
|
| 86 |
| Metric | Value |
|
| 87 |
|---|---|
|
| 88 |
-
| Train loss | 0.
|
| 89 |
-
| Eval loss | 0.
|
| 90 |
| Eval accuracy | 0.9395 |
|
| 91 |
-
|
|
| 92 |
|
| 93 |
## Experiment Tracking
|
| 94 |
|
|
@@ -98,7 +110,7 @@ Trained on `dair-ai/emotion` — 4 epochs.
|
|
| 98 |
|
| 99 |
| Component | Tool |
|
| 100 |
|---|---|
|
| 101 |
-
| Compute | Tesla T4
|
| 102 |
-
|
|
| 103 |
-
|
|
| 104 |
| Orchestration | The Founder |
|
|
|
|
| 4 |
license: apache-2.0
|
| 5 |
tags:
|
| 6 |
- text-classification
|
|
|
|
|
|
|
| 7 |
- the-founder
|
| 8 |
+
- emotion
|
| 9 |
+
- 6-class
|
| 10 |
datasets:
|
| 11 |
- dair-ai/emotion
|
| 12 |
metrics:
|
|
|
|
| 18 |
results:
|
| 19 |
- task:
|
| 20 |
type: text-classification
|
| 21 |
+
name: Emotion Classification
|
| 22 |
dataset:
|
| 23 |
+
name: Emotion
|
| 24 |
type: dair-ai/emotion
|
| 25 |
split: validation
|
| 26 |
metrics:
|
|
|
|
| 28 |
value: 0.9395
|
| 29 |
name: Accuracy
|
| 30 |
- type: loss
|
| 31 |
+
value: 0.2680
|
| 32 |
name: Eval Loss
|
| 33 |
---
|
| 34 |
|
|
|
|
| 38 |
|
| 39 |
## Model Description
|
| 40 |
|
| 41 |
+
This is a fine-tuned version of `roberta-base` on the `dair-ai/emotion` dataset for Emotion Classification.
|
| 42 |
+
|
| 43 |
+
Training was orchestrated end-to-end by **The Founder** — a personal ML agent that handles research, compute scheduling, experiment tracking, and artifact management autonomously using Kaggle (Tesla T4), Weights & Biases, and HuggingFace Hub.
|
| 44 |
|
| 45 |
## Model Details
|
| 46 |
|
| 47 |
| Property | Value |
|
| 48 |
|---|---|
|
| 49 |
| Base model | roberta-base |
|
| 50 |
+
| Fine-tuned on | dair-ai/emotion |
|
| 51 |
+
| Task | Emotion Classification |
|
| 52 |
| Epochs | 4 |
|
| 53 |
| Batch size | 32 |
|
| 54 |
| Learning rate | 2e-05 |
|
| 55 |
| GPU | Tesla T4 (Kaggle) |
|
| 56 |
+
| Train loss | 0.2018 |
|
| 57 |
+
| Eval loss | 0.2680 |
|
| 58 |
| Eval accuracy | 0.9395 |
|
| 59 |
+
| Duration | 21.3 min |
|
| 60 |
|
| 61 |
## How to Get Started
|
| 62 |
|
| 63 |
```python
|
| 64 |
from transformers import pipeline
|
| 65 |
clf = pipeline("text-classification", model="zanesmit29/founder-roberta-emotion-v1")
|
| 66 |
+
clf("Your input text here")
|
| 67 |
```
|
| 68 |
|
| 69 |
## Uses
|
| 70 |
|
| 71 |
+
### Direct Use
|
| 72 |
+
This model is suitable for Emotion Classification tasks in English. It can be used out-of-the-box with the Transformers pipeline API.
|
| 73 |
+
|
| 74 |
+
### Out-of-Scope Use
|
| 75 |
+
This model was trained on a specific dataset and may not generalise to all domains or languages.
|
| 76 |
+
It should not be used to make high-stakes automated decisions without human review.
|
| 77 |
+
Performance on out-of-distribution data (e.g. non-English text, domain-specific jargon) is not guaranteed.
|
| 78 |
|
| 79 |
## Training Details
|
| 80 |
|
| 81 |
### Data
|
| 82 |
+
Trained on `dair-ai/emotion` — 4 epochs, no additional preprocessing beyond standard tokenization.
|
| 83 |
|
| 84 |
### Hyperparameters
|
| 85 |
|
| 86 |
+
| Hyperparameter | Value |
|
| 87 |
|---|---|
|
| 88 |
+
| Learning rate | 2e-05 |
|
| 89 |
+
| Batch size | 32 |
|
| 90 |
| Epochs | 4 |
|
| 91 |
+
| Optimizer | AdamW |
|
| 92 |
+
| LR scheduler | Linear with warmup |
|
| 93 |
+
| Max sequence length | 128 |
|
| 94 |
| fp16 | true |
|
| 95 |
|
| 96 |
## Results
|
| 97 |
|
| 98 |
| Metric | Value |
|
| 99 |
|---|---|
|
| 100 |
+
| Train loss | 0.2018 |
|
| 101 |
+
| Eval loss | 0.2680 |
|
| 102 |
| Eval accuracy | 0.9395 |
|
| 103 |
+
| Duration | 21.3 min |
|
| 104 |
|
| 105 |
## Experiment Tracking
|
| 106 |
|
|
|
|
| 110 |
|
| 111 |
| Component | Tool |
|
| 112 |
|---|---|
|
| 113 |
+
| Compute | Kaggle (Tesla T4) |
|
| 114 |
+
| Experiment tracking | Weights & Biases |
|
| 115 |
+
| Artifact storage | HuggingFace Hub |
|
| 116 |
| Orchestration | The Founder |
|