gorkaartola commited on
Commit
a65e727
1 Parent(s): 90f9b75

Upload options.py

Browse files
Files changed (1) hide show
  1. options.py +38 -0
options.py ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ test_datasets = {
2
+ 'gorkaartola/SC-ZS-test_AURORA-Gold-SDG_True-Positives-and-False-Positives' : '0',
3
+ }
4
+
5
+ models = {
6
+ 'joeddav/xlm-roberta-large-xnli' : '0',
7
+ }
8
+
9
+ queries = {
10
+ 'SDG_Titles.csv' : '0',
11
+ 'SDG_Headlines.csv' : '1',
12
+ 'SDG_Subjects.csv' : '2',
13
+ 'SDG_Targets.csv' : '3',
14
+ 'SDG_Numbers.csv' : '4',
15
+ }
16
+
17
+ prompts = {
18
+ '' : '0',
19
+ 'This is ': '1',
20
+ 'The subject is ': '2',
21
+ 'The Sustainable Development Goal is ': '3',
22
+ }
23
+
24
+ metrics = {
25
+ 'gorkaartola/metric_for_tp_fp_samples': '0',
26
+ }
27
+
28
+ prediction_strategy_options = {
29
+ "argmax" : ["argmax_max"],
30
+ "threshold 0.05" : ["threshold", 0.05],
31
+ "threshold 0.25" : ["threshold", 0.25],
32
+ "threshold 0.5" : ["threshold", 0.5],
33
+ "threshold 0.75" : ["threshold", 0.75],
34
+ "topk 9" : ["topk", 9],
35
+ "topk 7" : ["topk", 7],
36
+ "topk 5" : ["topk", 5],
37
+ "topk 3" : ["topk", 3],
38
+ }