raj-p commited on
Commit
6ef0ed4
1 Parent(s): a3e6278

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -12
README.md CHANGED
@@ -8,26 +8,19 @@ model-index:
8
  results: []
9
  ---
10
 
11
- <!-- This model card has been generated automatically according to the information Keras had access to. You should
12
- probably proofread and complete it, then remove this comment. -->
13
-
14
  # bert-base-uncased-finetuned-glue-sst2
15
 
16
- This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on an unknown dataset.
17
- It achieves the following results on the evaluation set:
18
 
 
19
 
20
  ## Model description
21
 
22
- More information needed
23
-
24
- ## Intended uses & limitations
25
-
26
- More information needed
27
 
28
  ## Training and evaluation data
29
 
30
- More information needed
31
 
32
  ## Training procedure
33
 
@@ -39,7 +32,8 @@ The following hyperparameters were used during training:
39
 
40
  ### Training results
41
 
42
-
 
43
 
44
  ### Framework versions
45
 
 
8
  results: []
9
  ---
10
 
 
 
 
11
  # bert-base-uncased-finetuned-glue-sst2
12
 
13
+ Use for **sentiment analysis**.
 
14
 
15
+ This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on the [glue sst2 dataset](https://huggingface.co/datasets/glue/viewer/sst2). The model achieves `91.39%` accuracy on the validation dataset.
16
 
17
  ## Model description
18
 
19
+ [bert-base-uncased](https://huggingface.co/bert-base-uncased) is a pretrained English language model. `bert-base-uncased-finetuned-glue-sst2` adds a 2-class classification head for predicting `positive` and `negative` sentiment.
 
 
 
 
20
 
21
  ## Training and evaluation data
22
 
23
+ The model has been trained on 10K training samples, even though the [glue sst2 dataset](https://huggingface.co/datasets/glue/viewer/sst2) contains 67.3K samples. This was done to decrease training time.
24
 
25
  ## Training procedure
26
 
 
32
 
33
  ### Training results
34
 
35
+ - Accuracy (training): `94.33%`
36
+ - Accuracy (validation): `91.39%`
37
 
38
  ### Framework versions
39