zanesmit29 commited on
Commit
399cc79
·
verified ·
1 Parent(s): 8b279c0

Add canonical model card — v6 Kaggle T4, 93.95% accuracy

Browse files
Files changed (1) hide show
  1. README.md +34 -22
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: Text Classification
22
  dataset:
23
- name: dair-ai/emotion
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.2677
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
- Fine-tuned `roberta-base` on `dair-ai/emotion` for 6-class emotion classification (sadness, joy, love, anger, fear, surprise).
 
 
42
 
43
  ## Model Details
44
 
45
  | Property | Value |
46
  |---|---|
47
  | Base model | roberta-base |
48
- | Dataset | dair-ai/emotion |
 
49
  | Epochs | 4 |
50
  | Batch size | 32 |
51
  | Learning rate | 2e-05 |
52
  | GPU | Tesla T4 (Kaggle) |
53
- | Train loss | 0.7915 |
54
- | Eval loss | 0.2677 |
55
  | Eval accuracy | 0.9395 |
56
- | Duration | 12.3 min |
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("I am so happy today!")
64
  ```
65
 
66
  ## Uses
67
 
68
- Emotion detection across sadness, joy, love, anger, fear, surprise.
 
 
 
 
 
 
69
 
70
  ## Training Details
71
 
72
  ### Data
73
- Trained on `dair-ai/emotion` — 4 epochs.
74
 
75
  ### Hyperparameters
76
 
77
- | Param | Value |
78
  |---|---|
79
- | LR | 2e-05 |
80
- | Batch | 32 |
81
  | Epochs | 4 |
 
 
 
82
  | fp16 | true |
83
 
84
  ## Results
85
 
86
  | Metric | Value |
87
  |---|---|
88
- | Train loss | 0.7915 |
89
- | Eval loss | 0.2677 |
90
  | Eval accuracy | 0.9395 |
91
- | Verdict | CONFIRMED |
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 (Kaggle) |
102
- | Tracking | Weights & Biases |
103
- | Artifacts | HuggingFace Hub |
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 |