MLRC_Bench / src /utils /task_mapping.py
Armeddinosaur's picture
Updating metrics
678bdbb
raw
history blame
1.16 kB
"""
Task name mapping utilities for displaying user-friendly task names.
"""
# Mapping from original task names to display names
task_display_names = {
"Perception Temporal Action Loc": "Temporal Action Localisation",
"Llm Merging": "LLM Merging",
"Meta Learning": "Meta Learning",
"Product Recommendation": "Next Product Recommendation",
"Weather Forcast": "Rainfall Prediction",
"Machine Unlearning": "Machine Unlearning",
"Backdoor Trigger Recovery": "Backdoor Trigger Recovery"
}
def get_display_name(task_name):
"""
Get the display name for a task
Args:
task_name (str): The original task name
Returns:
str: The display name for the task
"""
return task_display_names.get(task_name, task_name)
def get_original_name(display_name):
"""
Get the original task name for a display name
Args:
display_name (str): The display name
Returns:
str: The original task name
"""
# Create a reverse mapping
reverse_mapping = {v: k for k, v in task_display_names.items()}
return reverse_mapping.get(display_name, display_name)