rwillh11 commited on
Commit
e25afd9
·
verified ·
1 Parent(s): f37a2c9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -2
README.md CHANGED
@@ -9,9 +9,8 @@ This model takes the various social groups that might be mentioned in political
9
 
10
  See Dolinsky et al (2025) for more information on social group categories and Horne et al (2025) for details on training, relation to other models, and use cases.
11
 
12
- 📊 Evaluation Results: {'eval_loss': 0.024197660386562347, 'eval_accuracy': 0.994262946025139, 'eval_f1': 0.8552677029360967, 'eval_precision': 0.8657342657342657, 'eval_recall': 0.8450511945392492, 'eval_runtime': 5.9184, 'eval_samples_per_second': 232.831, 'eval_steps_per_second': 29.231, 'epoch': 29.914368650217707}
13
 
14
- We found optimal hyper-parameters to be:
15
 
16
  LEARNING_RATE = 1.9432557585419205e-05
17
  WEIGHT_DECAY = 0.11740203810285466
 
9
 
10
  See Dolinsky et al (2025) for more information on social group categories and Horne et al (2025) for details on training, relation to other models, and use cases.
11
 
12
+ 📊 Evaluation Results: {'eval_loss': 0.023458, 'eval_accuracy': 0.994933, 'eval_f1': 0.894393, 'eval_precision': 0.0.897170, 'eval_recall': 0.891632, 'eval_runtime': 5.9184, 'eval_samples_per_second': 232.831, 'eval_steps_per_second': 29.231, 'epoch': 29.914368650217707}
13
 
 
14
 
15
  LEARNING_RATE = 1.9432557585419205e-05
16
  WEIGHT_DECAY = 0.11740203810285466