gorkaartola commited on
Commit
d81f1d7
1 Parent(s): cb3be1a

Upload options.py

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