eclec commited on
Commit
774cdc3
·
1 Parent(s): d1cd8d8

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -27
README.md CHANGED
@@ -1,12 +1,6 @@
1
  ---
2
- base_model: allenai/longformer-large-4096
3
  tags:
4
  - generated_from_trainer
5
- metrics:
6
- - accuracy
7
- - f1
8
- - precision
9
- - recall
10
  model-index:
11
  - name: patentClassfication2
12
  results: []
@@ -17,13 +11,7 @@ should probably proofread and complete it, then remove this comment. -->
17
 
18
  # patentClassfication2
19
 
20
- This model is a fine-tuned version of [allenai/longformer-large-4096](https://huggingface.co/allenai/longformer-large-4096) on the None dataset.
21
- It achieves the following results on the evaluation set:
22
- - Loss: 0.6395
23
- - Accuracy: 0.645
24
- - F1: 0.6764
25
- - Precision: 0.6214
26
- - Recall: 0.742
27
 
28
  ## Model description
29
 
@@ -42,25 +30,17 @@ More information needed
42
  ### Training hyperparameters
43
 
44
  The following hyperparameters were used during training:
45
- - learning_rate: 7.224533222600416e-06
46
  - train_batch_size: 8
47
  - eval_batch_size: 8
48
- - seed: 3
 
 
49
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
50
  - lr_scheduler_type: cosine
51
  - lr_scheduler_warmup_ratio: 0.1
52
- - lr_scheduler_warmup_steps: 371
53
- - num_epochs: 4
54
-
55
- ### Training results
56
-
57
- | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Precision | Recall |
58
- |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
59
- | 0.6917 | 1.0 | 500 | 0.6714 | 0.595 | 0.64 | 0.576 | 0.72 |
60
- | 0.6577 | 2.0 | 1000 | 0.6395 | 0.645 | 0.6764 | 0.6214 | 0.742 |
61
- | 0.536 | 3.0 | 1500 | 0.6535 | 0.675 | 0.6531 | 0.7002 | 0.612 |
62
- | 0.4025 | 4.0 | 2000 | 0.6808 | 0.686 | 0.6879 | 0.6838 | 0.692 |
63
-
64
 
65
  ### Framework versions
66
 
 
1
  ---
 
2
  tags:
3
  - generated_from_trainer
 
 
 
 
 
4
  model-index:
5
  - name: patentClassfication2
6
  results: []
 
11
 
12
  # patentClassfication2
13
 
14
+ This model was trained from scratch on the None dataset.
 
 
 
 
 
 
15
 
16
  ## Model description
17
 
 
30
  ### Training hyperparameters
31
 
32
  The following hyperparameters were used during training:
33
+ - learning_rate: 3.250956097988812e-05
34
  - train_batch_size: 8
35
  - eval_batch_size: 8
36
+ - seed: 48
37
+ - gradient_accumulation_steps: 8
38
+ - total_train_batch_size: 64
39
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
40
  - lr_scheduler_type: cosine
41
  - lr_scheduler_warmup_ratio: 0.1
42
+ - lr_scheduler_warmup_steps: 495
43
+ - num_epochs: 2
 
 
 
 
 
 
 
 
 
 
44
 
45
  ### Framework versions
46