Spaces:
Runtime error
Runtime error
| import json | |
| import os | |
| import glob | |
| import numpy as np | |
| def convert_model_name(model): | |
| name = { | |
| "T5": "google-flan-t5-large", | |
| "UL2": "google-flan-ul2", | |
| "Vicuna": "vicuna-13b", | |
| "ChatGPT": "chatgpt", | |
| } | |
| return name[model] | |
| def retrieve_transfer(source, target, attack, shot): | |
| source = convert_model_name(source) | |
| target = convert_model_name(target) | |
| attack = attack.lower() | |
| file_dir = "./results_transfer/"+source+"_"+target+"/"+attack+"_"+str(shot)+"_shot.json" | |
| with open(file_dir, 'r', encoding='utf-8') as f: | |
| data = json.load(f) | |
| return data | |