bergum commited on
Commit
d92fa23
1 Parent(s): aaf180c

Update readme with links to training routine

Browse files
Files changed (1) hide show
  1. README.md +19 -6
README.md CHANGED
@@ -1,28 +1,41 @@
1
  ---
2
  license: apache-2.0
3
- tags:
4
- - generated_from_trainer
5
  datasets:
6
  - go_emotions
7
 
 
 
8
  model-index:
9
- - name: xtremedistil-l6-h384-go-emotion
10
  results:
11
  - task:
12
- name: Text Classification
13
  type: multi_label_classification
14
  dataset:
15
  name: go_emotions
16
  type: emotion
17
  args: default
18
-
 
 
 
 
19
  ---
20
  # xtremedistil-l6-h384-go-emotion
21
- This model is a fine-tuned version of [microsoft/xtremedistil-l6-h384-uncased](https://huggingface.co/microsoft/xtremedistil-l6-h384-uncased) on the go_emotions dataset.
 
 
 
 
 
 
 
22
 
23
 
24
  ### Training hyperparameters
 
25
  - learning_rate=3e-05
 
26
  <pre>
27
  Num examples = 211225
28
  Num Epochs = 4
1
  ---
2
  license: apache-2.0
 
 
3
  datasets:
4
  - go_emotions
5
 
6
+ metrics:
7
+ - accuracy
8
  model-index:
9
+ - name: xtremedistil-emotion
10
  results:
11
  - task:
12
+ name: Multi Label Text Classification
13
  type: multi_label_classification
14
  dataset:
15
  name: go_emotions
16
  type: emotion
17
  args: default
18
+ metrics:
19
+ - name: Accuracy
20
+ type: accuracy
21
+ value: NaN
22
+
23
  ---
24
  # xtremedistil-l6-h384-go-emotion
25
+ This model is a fine-tuned version of [microsoft/xtremedistil-l6-h384-uncased](https://huggingface.co/microsoft/xtremedistil-l6-h384-uncased) on the
26
+ [go_emotions dataset](https://huggingface.co/datasets/go_emotions).
27
+
28
+ [![Training Notebook](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/jobergum/emotion/blob/main/TrainGoEmotions.ipynb)
29
+
30
+ This model is deployed to [aiserv.cloud](https://aiserv.cloud/) for live demo of in-Browser ML inference
31
+ See [https://github.com/jobergum/browser-ml-inference](https://github.com/jobergum/browser-ml-inference)
32
+
33
 
34
 
35
  ### Training hyperparameters
36
+ - batch size 128
37
  - learning_rate=3e-05
38
+ - epocs 4
39
  <pre>
40
  Num examples = 211225
41
  Num Epochs = 4