Bencode92 commited on
Commit
9dc9ede
·
1 Parent(s): 2d80731

🔄 Incremental importance | Acc: 0.714, F1: 0.689

Browse files
README.md CHANGED
@@ -21,19 +21,19 @@ Fine-tuned FinBERT model for financial importance analysis in TradePulse.
21
 
22
  ## Performance
23
 
24
- *Last training: 2025-07-24 17:35*
25
  *Dataset: `news_20250724.csv` (125 samples)*
26
 
27
  | Metric | Value |
28
  |--------|-------|
29
- | Loss | 0.8915 |
30
- | Accuracy | 0.7188 |
31
- | F1 Score | 0.7226 |
32
 
33
- | F1 Macro | 0.7226 |
34
 
35
- | Precision | 0.7324 |
36
- | Recall | 0.7188 |
37
 
38
  ## Training Details
39
 
@@ -65,4 +65,4 @@ predictions = outputs.logits.softmax(dim=-1)
65
  ## Model Card Authors
66
 
67
  - TradePulse ML Team
68
- - Auto-generated on 2025-07-24 17:35:40
 
21
 
22
  ## Performance
23
 
24
+ *Last training: 2025-07-24 21:27*
25
  *Dataset: `news_20250724.csv` (125 samples)*
26
 
27
  | Metric | Value |
28
  |--------|-------|
29
+ | Loss | 0.9156 |
30
+ | Accuracy | 0.7500 |
31
+ | F1 Score | 0.7520 |
32
 
33
+ | F1 Macro | 0.7520 |
34
 
35
+ | Precision | 0.7555 |
36
+ | Recall | 0.7500 |
37
 
38
  ## Training Details
39
 
 
65
  ## Model Card Authors
66
 
67
  - TradePulse ML Team
68
+ - Auto-generated on 2025-07-24 21:27:54
checkpoint-48/model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9928f9028b14eb5805cea3810c00e510c4455a95477d0ead86968211031353a4
3
  size 439039996
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cf91ca40087521cccbf74de6a5820250349be0eb9b742a9ddb4f5d6f6da15de7
3
  size 439039996
checkpoint-48/trainer_state.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "best_metric": 0.7225877192982456,
3
  "best_model_checkpoint": "hf-importance-production/checkpoint-48",
4
  "epoch": 2.0,
5
  "eval_steps": 500,
@@ -10,60 +10,60 @@
10
  "log_history": [
11
  {
12
  "epoch": 0.4166666666666667,
13
- "grad_norm": 41.93825149536133,
14
  "learning_rate": 4.347826086956522e-06,
15
- "loss": 1.3382,
16
  "step": 10
17
  },
18
  {
19
  "epoch": 0.8333333333333334,
20
- "grad_norm": 64.2195816040039,
21
  "learning_rate": 8.695652173913044e-06,
22
- "loss": 1.6985,
23
  "step": 20
24
  },
25
  {
26
  "epoch": 1.0,
27
- "eval_accuracy": 0.71875,
28
- "eval_f1": 0.7022058823529411,
29
- "eval_f1_macro": 0.46274509803921565,
30
- "eval_loss": 1.1443504095077515,
31
- "eval_precision": 0.7364583333333333,
32
- "eval_precision_macro": 0.5177777777777778,
33
- "eval_recall": 0.71875,
34
- "eval_recall_macro": 0.4515151515151515,
35
- "eval_runtime": 17.6583,
36
- "eval_samples_per_second": 1.812,
37
- "eval_steps_per_second": 0.453,
38
  "step": 24
39
  },
40
  {
41
  "epoch": 1.25,
42
- "grad_norm": 50.58437728881836,
43
  "learning_rate": 7.2000000000000005e-06,
44
- "loss": 1.2258,
45
  "step": 30
46
  },
47
  {
48
  "epoch": 1.6666666666666665,
49
- "grad_norm": 18.62161636352539,
50
  "learning_rate": 3.2000000000000003e-06,
51
- "loss": 0.5873,
52
  "step": 40
53
  },
54
  {
55
  "epoch": 2.0,
56
- "eval_accuracy": 0.71875,
57
- "eval_f1": 0.7225877192982456,
58
- "eval_f1_macro": 0.48538011695906436,
59
- "eval_loss": 0.8914567828178406,
60
- "eval_precision": 0.7323717948717949,
61
- "eval_precision_macro": 0.48290598290598297,
62
- "eval_recall": 0.71875,
63
- "eval_recall_macro": 0.49242424242424243,
64
- "eval_runtime": 17.6172,
65
- "eval_samples_per_second": 1.816,
66
- "eval_steps_per_second": 0.454,
67
  "step": 48
68
  }
69
  ],
 
1
  {
2
+ "best_metric": 0.7519509476031214,
3
  "best_model_checkpoint": "hf-importance-production/checkpoint-48",
4
  "epoch": 2.0,
5
  "eval_steps": 500,
 
10
  "log_history": [
11
  {
12
  "epoch": 0.4166666666666667,
13
+ "grad_norm": 33.77495574951172,
14
  "learning_rate": 4.347826086956522e-06,
15
+ "loss": 0.2528,
16
  "step": 10
17
  },
18
  {
19
  "epoch": 0.8333333333333334,
20
+ "grad_norm": 41.324256896972656,
21
  "learning_rate": 8.695652173913044e-06,
22
+ "loss": 0.3631,
23
  "step": 20
24
  },
25
  {
26
  "epoch": 1.0,
27
+ "eval_accuracy": 0.6875,
28
+ "eval_f1": 0.6842334494773519,
29
+ "eval_f1_macro": 0.45063879210220675,
30
+ "eval_loss": 0.9002108573913574,
31
+ "eval_precision": 0.6824404761904761,
32
+ "eval_precision_macro": 0.45396825396825397,
33
+ "eval_recall": 0.6875,
34
+ "eval_recall_macro": 0.4484848484848485,
35
+ "eval_runtime": 17.967,
36
+ "eval_samples_per_second": 1.781,
37
+ "eval_steps_per_second": 0.445,
38
  "step": 24
39
  },
40
  {
41
  "epoch": 1.25,
42
+ "grad_norm": 40.574581146240234,
43
  "learning_rate": 7.2000000000000005e-06,
44
+ "loss": 0.2537,
45
  "step": 30
46
  },
47
  {
48
  "epoch": 1.6666666666666665,
49
+ "grad_norm": 4.176632881164551,
50
  "learning_rate": 3.2000000000000003e-06,
51
+ "loss": 0.1811,
52
  "step": 40
53
  },
54
  {
55
  "epoch": 2.0,
56
+ "eval_accuracy": 0.75,
57
+ "eval_f1": 0.7519509476031214,
58
+ "eval_f1_macro": 0.5053883314752879,
59
+ "eval_loss": 0.9156143069267273,
60
+ "eval_precision": 0.7554824561403508,
61
+ "eval_precision_macro": 0.5029239766081871,
62
+ "eval_recall": 0.75,
63
+ "eval_recall_macro": 0.5090909090909091,
64
+ "eval_runtime": 17.8939,
65
+ "eval_samples_per_second": 1.788,
66
+ "eval_steps_per_second": 0.447,
67
  "step": 48
68
  }
69
  ],
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9928f9028b14eb5805cea3810c00e510c4455a95477d0ead86968211031353a4
3
  size 439039996
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cf91ca40087521cccbf74de6a5820250349be0eb9b742a9ddb4f5d6f6da15de7
3
  size 439039996