gorkaartola's picture
Upload options.py
d81f1d7
raw history blame
No virus
741 Bytes
models = {
'joeddav/xlm-roberta-large-xnli' : '0',
}
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',
}
metrics = {
'gorkaartola/metric_for_tp_fp_samples': '0',
}
prediction_strategy_options = {
"argmax_max" : ["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],
}