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