File size: 664 Bytes
25bd958 c5e55e7 1fae342 25bd958 ffa21be f62f9ff 25bd958 f62f9ff 25bd958 f62f9ff 25bd958 f62f9ff c5e55e7 1fae342 f62f9ff fed5855 ffa21be 47d01e3 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 | DATASETS_KEYWORDS = [
"train-size",
"test-size",
"c",
"csharp",
"python",
"java",
"javascript",
]
def get_datasets_nickname(dataset_name) -> str:
if "csharp" == dataset_name:
return "C#"
elif "python" == dataset_name:
return "Python"
elif "javascript" == dataset_name:
return "JavaScript"
elif "java" == dataset_name:
return "Java"
elif "train-size" == dataset_name:
return "Train\nSize"
elif "test-size" == dataset_name:
return "Test\nSize"
elif "c" == dataset_name:
return "C"
else:
raise ValueError(f"Dataset {dataset_name} not found")
|