Kayvane commited on
Commit
79fe463
1 Parent(s): 2f3bab4

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -36
README.md CHANGED
@@ -4,34 +4,9 @@ tags:
4
  - generated_from_trainer
5
  datasets:
6
  - consumer-finance-complaints
7
- metrics:
8
- - accuracy
9
- - f1
10
- - recall
11
- - precision
12
  model-index:
13
  - name: distilbert-base-uncased-wandb-week-3-complaints-classifier-1500
14
- results:
15
- - task:
16
- name: Text Classification
17
- type: text-classification
18
- dataset:
19
- name: consumer-finance-complaints
20
- type: consumer-finance-complaints
21
- args: default
22
- metrics:
23
- - name: Accuracy
24
- type: accuracy
25
- value: 0.8210081035625095
26
- - name: F1
27
- type: f1
28
- value: 0.8181860806077187
29
- - name: Recall
30
- type: recall
31
- value: 0.8210081035625095
32
- - name: Precision
33
- type: precision
34
- value: 0.8187295355668672
35
  ---
36
 
37
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -40,12 +15,6 @@ should probably proofread and complete it, then remove this comment. -->
40
  # distilbert-base-uncased-wandb-week-3-complaints-classifier-1500
41
 
42
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the consumer-finance-complaints dataset.
43
- It achieves the following results on the evaluation set:
44
- - Loss: 0.5521
45
- - Accuracy: 0.8210
46
- - F1: 0.8182
47
- - Recall: 0.8210
48
- - Precision: 0.8187
49
 
50
  ## Model description
51
 
@@ -76,10 +45,6 @@ The following hyperparameters were used during training:
76
 
77
  ### Training results
78
 
79
- | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Recall | Precision |
80
- |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:------:|:---------:|
81
- | 0.7063 | 0.82 | 2000 | 0.6899 | 0.7809 | 0.7544 | 0.7809 | 0.7646 |
82
- | 0.505 | 1.63 | 4000 | 0.5521 | 0.8210 | 0.8182 | 0.8210 | 0.8187 |
83
 
84
 
85
  ### Framework versions
 
4
  - generated_from_trainer
5
  datasets:
6
  - consumer-finance-complaints
 
 
 
 
 
7
  model-index:
8
  - name: distilbert-base-uncased-wandb-week-3-complaints-classifier-1500
9
+ results: []
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
10
  ---
11
 
12
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
15
  # distilbert-base-uncased-wandb-week-3-complaints-classifier-1500
16
 
17
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the consumer-finance-complaints dataset.
 
 
 
 
 
 
18
 
19
  ## Model description
20
 
 
45
 
46
  ### Training results
47
 
 
 
 
 
48
 
49
 
50
  ### Framework versions