Spaces:
Running
Running
Update src/populate.py
Browse files- 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
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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)
|