apugachev commited on
Commit
b112de6
·
1 Parent(s): f431d57

Upload README.md

Browse files
Files changed (1) hide show
  1. README.md +30 -0
README.md ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ Training parameters:
2
+
3
+ ```
4
+ model_args = ClassificationArgs()
5
+ model_args.max_seq_length = 512
6
+ model_args.train_batch_size = 12
7
+ model_args.eval_batch_size = 12
8
+ model_args.num_train_epochs = 5
9
+ model_args.evaluate_during_training = False
10
+ model_args.learning_rate = 1e-5
11
+ model_args.use_multiprocessing = False
12
+ model_args.fp16 = False
13
+ model_args.save_steps = -1
14
+ model_args.save_eval_checkpoints = False
15
+ model_args.no_cache = True
16
+ model_args.reprocess_input_data = True
17
+ model_args.overwrite_output_dir = True
18
+ ```
19
+
20
+ Evaluation on BoolQ Test Set:
21
+
22
+ | | Precision | Recall | F1-score |
23
+ |:------------:|:---------:|:------:|:--------:|
24
+ | 0 | 0.82 | 0.80 | 0.81 |
25
+ | 1 | 0.88 | 0.89 | 0.88 |
26
+ | accuracy | | | 0.86 |
27
+ | macro avg | 0.85 | 0.84 | 0.85 |
28
+ | weighted avg | 0.86 | 0.86 | 0.86 |
29
+
30
+ ROC AUC Score: 0.844