LucasLima commited on
Commit
31b5bf1
·
verified ·
1 Parent(s): 316d130

Update src/populate.py

Browse files
Files changed (1) hide show
  1. src/populate.py +14 -1
src/populate.py CHANGED
@@ -42,7 +42,20 @@ def get_leaderboard_df(results_path: str = None, requests_path: str = None, cols
42
  df[avg_col_name] = np.nan
43
  print(f"Nenhuma coluna encontrada para {area_name}, definindo média como NaN.")
44
 
45
- # Calcular Média Geral (agora baseada nas médias das áreas)
 
 
 
 
 
 
 
 
 
 
 
 
 
46
  avg_area_cols = [col for col in AREA_AVG_COLUMN_MAP.values() if col in df.columns]
47
  if avg_area_cols:
48
  df[AutoEvalColumn.average.name] = df[avg_area_cols].mean(axis=1, skipna=True)
 
42
  df[avg_col_name] = np.nan
43
  print(f"Nenhuma coluna encontrada para {area_name}, definindo média como NaN.")
44
 
45
+ # Calcular Média PLUE
46
+ plue_avg_cols_to_consider = [
47
+ AREA_AVG_COLUMN_MAP[area]
48
+ for area in PLUE_GROUP_AREAS
49
+ if area in AREA_AVG_COLUMN_MAP and AREA_AVG_COLUMN_MAP[area] in df.columns
50
+ ]
51
+ if plue_avg_cols_to_consider:
52
+ df[AutoEvalColumn.plue_avg.name] = df[plue_avg_cols_to_consider].mean(axis=1, skipna=True)
53
+ print(f"Calculada Média PLUE usando colunas: {plue_avg_cols_to_consider}")
54
+ else:
55
+ df[AutoEvalColumn.plue_avg.name] = np.nan
56
+ print("Nenhuma coluna de média de área PLUE encontrada, definindo Média PLUE como NaN.")
57
+
58
+ # Calcular Média Geral (baseada nas médias de TODAS as áreas)
59
  avg_area_cols = [col for col in AREA_AVG_COLUMN_MAP.values() if col in df.columns]
60
  if avg_area_cols:
61
  df[AutoEvalColumn.average.name] = df[avg_area_cols].mean(axis=1, skipna=True)