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

🔄 Incremental importance | Acc: 0.500, F1: 0.497

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 16:00*
25
- *Dataset: `news_20250724.csv` (190 samples)*
26
 
27
  | Metric | Value |
28
  |--------|-------|
29
- | Loss | 0.1901 |
30
- | Accuracy | 1.0000 |
31
- | F1 Score | 1.0000 |
32
 
33
- | F1 Macro | 1.0000 |
34
 
35
- | Precision | 1.0000 |
36
- | Recall | 1.0000 |
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 16:00:13
 
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
  ## Model Card Authors
66
 
67
  - TradePulse ML Team
68
+ - Auto-generated on 2025-07-24 17:35:40
checkpoint-48/model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4a21169b75795694e2eade74c7631920af5a7199ac0a38451ea6675feafcbd2c
3
  size 439039996
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9928f9028b14eb5805cea3810c00e510c4455a95477d0ead86968211031353a4
3
  size 439039996
checkpoint-48/trainer_state.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
- "best_metric": 1.0,
3
  "best_model_checkpoint": "hf-importance-production/checkpoint-48",
4
- "epoch": 1.0,
5
  "eval_steps": 500,
6
  "global_step": 48,
7
  "is_hyper_param_search": false,
@@ -9,51 +9,66 @@
9
  "is_world_process_zero": true,
10
  "log_history": [
11
  {
12
- "epoch": 0.20833333333333334,
13
- "grad_norm": 29.484235763549805,
14
- "learning_rate": 2.1276595744680853e-06,
15
- "loss": 0.139,
16
  "step": 10
17
  },
18
  {
19
- "epoch": 0.4166666666666667,
20
- "grad_norm": 0.19761385023593903,
21
- "learning_rate": 4.255319148936171e-06,
22
- "loss": 0.0851,
23
  "step": 20
24
  },
25
  {
26
- "epoch": 0.625,
27
- "grad_norm": 75.20068359375,
28
- "learning_rate": 6.382978723404256e-06,
29
- "loss": 0.2067,
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
30
  "step": 30
31
  },
32
  {
33
- "epoch": 0.8333333333333334,
34
- "grad_norm": 0.0050959475338459015,
35
- "learning_rate": 8.510638297872341e-06,
36
- "loss": 0.2117,
37
  "step": 40
38
  },
39
  {
40
- "epoch": 1.0,
41
- "eval_accuracy": 1.0,
42
- "eval_f1": 1.0,
43
- "eval_f1_macro": 1.0,
44
- "eval_loss": 0.19008873403072357,
45
- "eval_precision": 1.0,
46
- "eval_precision_macro": 1.0,
47
- "eval_recall": 1.0,
48
- "eval_recall_macro": 1.0,
49
- "eval_runtime": 0.5835,
50
- "eval_samples_per_second": 1.714,
51
- "eval_steps_per_second": 1.714,
52
  "step": 48
53
  }
54
  ],
55
  "logging_steps": 10,
56
- "max_steps": 96,
57
  "num_input_tokens_seen": 0,
58
  "num_train_epochs": 2,
59
  "save_steps": 500,
@@ -73,12 +88,12 @@
73
  "should_evaluate": false,
74
  "should_log": false,
75
  "should_save": true,
76
- "should_training_stop": false
77
  },
78
  "attributes": {}
79
  }
80
  },
81
- "total_flos": 49728435950592.0,
82
  "train_batch_size": 4,
83
  "trial_name": null,
84
  "trial_params": null
 
1
  {
2
+ "best_metric": 0.7225877192982456,
3
  "best_model_checkpoint": "hf-importance-production/checkpoint-48",
4
+ "epoch": 2.0,
5
  "eval_steps": 500,
6
  "global_step": 48,
7
  "is_hyper_param_search": false,
 
9
  "is_world_process_zero": true,
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
  ],
70
  "logging_steps": 10,
71
+ "max_steps": 48,
72
  "num_input_tokens_seen": 0,
73
  "num_train_epochs": 2,
74
  "save_steps": 500,
 
88
  "should_evaluate": false,
89
  "should_log": false,
90
  "should_save": true,
91
+ "should_training_stop": true
92
  },
93
  "attributes": {}
94
  }
95
  },
96
+ "total_flos": 48939095697408.0,
97
  "train_batch_size": 4,
98
  "trial_name": null,
99
  "trial_params": null
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4a21169b75795694e2eade74c7631920af5a7199ac0a38451ea6675feafcbd2c
3
  size 439039996
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9928f9028b14eb5805cea3810c00e510c4455a95477d0ead86968211031353a4
3
  size 439039996
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5d80d1e81c1a0fc229e86f25920b3a900302cd322f503a1e512596b0bb8d9a54
3
  size 5521
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:93be2cf0885ccccc46f3fa93fc94df6d8f955436f3fad71bd10c8ec497c8fae9
3
  size 5521