Spaces:
Runtime error
Runtime error
gorkaartola
commited on
Commit
•
d81f1d7
1
Parent(s):
cb3be1a
Upload options.py
Browse files- 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 |
+
}
|