LucasLima commited on
Commit
316d130
·
verified ·
1 Parent(s): 91ff1cc

Altração para Exibição do PLUE

Browse files
Files changed (1) hide show
  1. src/display/utils.py +14 -11
src/display/utils.py CHANGED
@@ -28,18 +28,21 @@ auto_eval_column_dict.append(["model", ColumnContent, ColumnContent("Model", "ma
28
  #Scores
29
  auto_eval_column_dict.append(["average", ColumnContent, ColumnContent("Média Geral ⬆️", "number", True)])
30
 
31
- # Adicionando colunas para as médias das áreas
32
- auto_eval_column_dict.append(["area_medica_avg", ColumnContent, ColumnContent("Área Médica", "number", True)])
33
- auto_eval_column_dict.append(["area_direito_avg", ColumnContent, ColumnContent("Área do Direito", "number", True)])
34
- auto_eval_column_dict.append(["provas_militares_avg", ColumnContent, ColumnContent("Provas Militares", "number", True)])
35
- auto_eval_column_dict.append(["computacao_avg", ColumnContent, ColumnContent("Computação", "number", True)])
36
- auto_eval_column_dict.append(["discurso_odio_avg", ColumnContent, ColumnContent("Discurso de Ódio", "number", True)])
37
- auto_eval_column_dict.append(["economia_contabilidade_avg", ColumnContent, ColumnContent("Economia e Contabilidade", "number", True)])
38
- auto_eval_column_dict.append(["semantica_inferencia_avg", ColumnContent, ColumnContent("Semântica e Inferência", "number", True)])
39
- auto_eval_column_dict.append(["multidisciplinar_avg", ColumnContent, ColumnContent("Multidisciplinar", "number", True)])
 
 
 
40
 
41
  for task in Tasks:
42
- auto_eval_column_dict.append([task.name, ColumnContent, ColumnContent(task.value.col_name, "number", False)]) # Mudar para False para não exibir por padrão na aba geral
43
  # Model information
44
  auto_eval_column_dict.append(["model_type", ColumnContent, ColumnContent("Type", "str", False)])
45
  auto_eval_column_dict.append(["architecture", ColumnContent, ColumnContent("Architecture", "str", False)])
@@ -66,7 +69,7 @@ AREA_DEFINITIONS = {
66
  "Multidisciplinar": [Tasks.ENEM, Tasks.BLUEX, Tasks.CNPU, Tasks.ENADE, Tasks.BNDES, Tasks.CACD_1, Tasks.CACD_2],
67
  }
68
 
69
- # Mapeamento dos nomes das áreas para as colunas de média correspondentes
70
  AREA_AVG_COLUMN_MAP = {
71
  "Área Médica": AutoEvalColumn.area_medica_avg.name,
72
  "Área do Direito": AutoEvalColumn.area_direito_avg.name,
 
28
  #Scores
29
  auto_eval_column_dict.append(["average", ColumnContent, ColumnContent("Média Geral ⬆️", "number", True)])
30
 
31
+ # Média específica do grupo PLUE
32
+ auto_eval_column_dict.append(["plue_avg", ColumnContent, ColumnContent("PLUE", "number", True)])
33
+
34
+ # Adicionando colunas para as médias das áreas (manter para cálculo e outras abas)
35
+ auto_eval_column_dict.append(["area_medica_avg", ColumnContent, ColumnContent("Área Médica", "number", False)]) # Exibido por padrão = False
36
+ auto_eval_column_dict.append(["area_direito_avg", ColumnContent, ColumnContent("Área do Direito", "number", False)]) # Exibido por padrão = False
37
+ auto_eval_column_dict.append(["provas_militares_avg", ColumnContent, ColumnContent("Provas Militares", "number", False)]) # Exibido por padrão = False
38
+ auto_eval_column_dict.append(["computacao_avg", ColumnContent, ColumnContent("Computação", "number", False)]) # Exibido por padrão = False
39
+ auto_eval_column_dict.append(["discurso_odio_avg", ColumnContent, ColumnContent("Discurso de Ódio", "number", True)]) # Manter True se fora do PLUE
40
+ auto_eval_column_dict.append(["economia_contabilidade_avg", ColumnContent, ColumnContent("Economia e Contabilidade", "number", True)]) # Manter True se fora do PLUE
41
+ auto_eval_column_dict.append(["semantica_inferencia_avg", ColumnContent, ColumnContent("Semântica e Inferência", "number", True)]) # Manter True se fora do PLUE
42
+ auto_eval_column_dict.append(["multidisciplinar_avg", ColumnContent, ColumnContent("Multidisciplinar", "number", False)]) # Exibido por padrão = False
43
 
44
  for task in Tasks:
45
+ auto_eval_column_dict.append([task.name, ColumnContent, ColumnContent(task.value.col_name, "number", False)])
46
  # Model information
47
  auto_eval_column_dict.append(["model_type", ColumnContent, ColumnContent("Type", "str", False)])
48
  auto_eval_column_dict.append(["architecture", ColumnContent, ColumnContent("Architecture", "str", False)])
 
69
  "Multidisciplinar": [Tasks.ENEM, Tasks.BLUEX, Tasks.CNPU, Tasks.ENADE, Tasks.BNDES, Tasks.CACD_1, Tasks.CACD_2],
70
  }
71
 
72
+ # Mapeamento dos nomes das áreas para as colunas de média correspondentes (Manter todos)
73
  AREA_AVG_COLUMN_MAP = {
74
  "Área Médica": AutoEvalColumn.area_medica_avg.name,
75
  "Área do Direito": AutoEvalColumn.area_direito_avg.name,