jkhan447 commited on
Commit
a56d2e0
·
1 Parent(s): ece50f7

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +18 -2
README.md CHANGED
@@ -4,9 +4,22 @@ tags:
4
  - generated_from_trainer
5
  datasets:
6
  - ag_news
 
 
7
  model-index:
8
  - name: sentiment-model-sample
9
- results: []
 
 
 
 
 
 
 
 
 
 
 
10
  ---
11
 
12
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -15,6 +28,9 @@ should probably proofread and complete it, then remove this comment. -->
15
  # sentiment-model-sample
16
 
17
  This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on the ag_news dataset.
 
 
 
18
 
19
  ## Model description
20
 
@@ -39,7 +55,7 @@ The following hyperparameters were used during training:
39
  - seed: 42
40
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
41
  - lr_scheduler_type: linear
42
- - num_epochs: 2
43
 
44
  ### Training results
45
 
 
4
  - generated_from_trainer
5
  datasets:
6
  - ag_news
7
+ metrics:
8
+ - accuracy
9
  model-index:
10
  - name: sentiment-model-sample
11
+ results:
12
+ - task:
13
+ name: Text Classification
14
+ type: text-classification
15
+ dataset:
16
+ name: ag_news
17
+ type: ag_news
18
+ args: default
19
+ metrics:
20
+ - name: Accuracy
21
+ type: accuracy
22
+ value: 0.9473684210526315
23
  ---
24
 
25
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
28
  # sentiment-model-sample
29
 
30
  This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on the ag_news dataset.
31
+ It achieves the following results on the evaluation set:
32
+ - Loss: 0.4887
33
+ - Accuracy: 0.9474
34
 
35
  ## Model description
36
 
 
55
  - seed: 42
56
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
57
  - lr_scheduler_type: linear
58
+ - num_epochs: 10
59
 
60
  ### Training results
61