gorkaartola's picture
Upload options.py
a72cebb
test_datasets = {
'gorkaartola/SC-ZS-test_AURORA-Gold-SDG_True-Positives-and-False-Positives' : '0',
}
metrics = {
'gorkaartola/metric_for_tp_fp_samples': '0',
}
models = {
'joeddav/xlm-roberta-large-xnli' : '0',
'facebook/bart-large-mnli' : '1',
}
queries = {
'gorkaartola/SDG_queries':
{
'SDG_Titles.csv' : '0',
'SDG_Headlines.csv' : '1',
'SDG_Subjects.csv' : '2',
'SDG_Targets.csv' : '3',
'SDG_Numbers.csv' : '4',
}
}
prompts = {
'' : '0',
'This is ': '1',
'The subject is ': '2',
'The Sustainable Development Goal is ': '3',
}
prediction_strategy_options = {
"argmax" : ["argmax_max"],
"threshold 0.05" : ["threshold", 0.05],
"threshold 0.25" : ["threshold", 0.25],
"threshold 0.5" : ["threshold", 0.5],
"threshold 0.75" : ["threshold", 0.75],
"topk 9" : ["topk", 9],
"topk 7" : ["topk", 7],
"topk 5" : ["topk", 5],
"topk 3" : ["topk", 3],
}