gorkaartola commited on
Commit
6809db0
1 Parent(s): 5cd06dc

Upload options.py

Browse files
Files changed (1) hide show
  1. options.py +6 -5
options.py CHANGED
@@ -1,11 +1,16 @@
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',
@@ -21,10 +26,6 @@ prompts = {
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],
 
1
  test_datasets = {
2
  'gorkaartola/SC-ZS-test_AURORA-Gold-SDG_True-Positives-and-False-Positives' : '0',
3
  }
4
+
5
+ metrics = {
6
+ 'gorkaartola/metric_for_tp_fp_samples': '0',
7
+ }
8
 
9
  models = {
10
  'joeddav/xlm-roberta-large-xnli' : '0',
11
+ 'facebook/bart-large-mnli' : '1',
12
  }
13
+
14
  queries = {
15
  'SDG_Titles.csv' : '0',
16
  'SDG_Headlines.csv' : '1',
 
26
  'The Sustainable Development Goal is ': '3',
27
  }
28
 
 
 
 
 
29
  prediction_strategy_options = {
30
  "argmax" : ["argmax_max"],
31
  "threshold 0.05" : ["threshold", 0.05],