jasonshaoshun commited on
Commit
7ed00c5
·
1 Parent(s): 53c7136
Files changed (1) hide show
  1. src/display/utils.py +4 -1
src/display/utils.py CHANGED
@@ -118,6 +118,7 @@ BENCHMARK_COLS_MIB_CAUSALGRAPH = []
118
  # ColumnContent,
119
  # ColumnContent(col_name, "number", True)
120
  # ])
 
121
  auto_eval_column_dict_mib_causalgraph = []
122
 
123
  # Method name column
@@ -126,10 +127,12 @@ auto_eval_column_dict_mib_causalgraph.append(["method", ColumnContent, ColumnCon
126
  # For each model-task-intervention combination
127
  for task in TasksMib_Causalgraph:
128
  for model in task.value.models:
 
129
  for layer in task.value.layers:
130
  for intervention in task.value.interventions:
131
  for counterfactual in task.value.counterfactuals:
132
- col_name = f"layer{layer}_{intervention}_{counterfactual}"
 
133
  field_name = col_name.lower()
134
  auto_eval_column_dict_mib_causalgraph.append([
135
  field_name,
 
118
  # ColumnContent,
119
  # ColumnContent(col_name, "number", True)
120
  # ])
121
+
122
  auto_eval_column_dict_mib_causalgraph = []
123
 
124
  # Method name column
 
127
  # For each model-task-intervention combination
128
  for task in TasksMib_Causalgraph:
129
  for model in task.value.models:
130
+ model_name = model.lower() # Convert model name to lowercase
131
  for layer in task.value.layers:
132
  for intervention in task.value.interventions:
133
  for counterfactual in task.value.counterfactuals:
134
+ # Include model name in the column name
135
+ col_name = f"{model_name}_layer{layer}_{intervention}_{counterfactual}"
136
  field_name = col_name.lower()
137
  auto_eval_column_dict_mib_causalgraph.append([
138
  field_name,