LucasLima commited on
Commit
efee332
·
verified ·
1 Parent(s): d961598

Update src/populate.py

Browse files
Files changed (1) hide show
  1. src/populate.py +4 -2
src/populate.py CHANGED
@@ -49,7 +49,8 @@ def get_leaderboard_df(results_path: str = None, requests_path: str = None, cols
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
@@ -58,7 +59,8 @@ def get_leaderboard_df(results_path: str = None, requests_path: str = None, cols
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)
 
62
  print(f"Calculada Média Geral usando colunas: {avg_area_cols}")
63
  else:
64
  df[AutoEvalColumn.average.name] = np.nan
 
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
+ # Substitui 0 por NaN antes de calcular a média PLUE
53
+ df[AutoEvalColumn.plue_avg.name] = df[plue_avg_cols_to_consider].replace(0, np.nan).mean(axis=1, skipna=True)
54
  print(f"Calculada Média PLUE usando colunas: {plue_avg_cols_to_consider}")
55
  else:
56
  df[AutoEvalColumn.plue_avg.name] = np.nan
 
59
  # Calcular Média Geral (baseada nas médias de TODAS as áreas)
60
  avg_area_cols = [col for col in AREA_AVG_COLUMN_MAP.values() if col in df.columns]
61
  if avg_area_cols:
62
+ # Substitui 0 por NaN antes de calcular a média Geral
63
+ df[AutoEvalColumn.average.name] = df[avg_area_cols].replace(0, np.nan).mean(axis=1, skipna=True)
64
  print(f"Calculada Média Geral usando colunas: {avg_area_cols}")
65
  else:
66
  df[AutoEvalColumn.average.name] = np.nan