Madhavan Iyengar commited on
Commit
942f95c
1 Parent(s): ee55dfa

use only accuracy metric

Browse files
Files changed (1) hide show
  1. src/display/utils.py +3 -2
src/display/utils.py CHANGED
@@ -19,7 +19,8 @@ class ColumnContent:
19
  never_hidden: bool = False
20
 
21
  # Dynamic fields based on new data schema
22
- metrics = ['precision', 'recall', 'f1_score', 'accuracy', 'yes_percent']
 
23
  auto_eval_column_dict = [
24
  ["model_name", ColumnContent, ColumnContent("Model Name", "markdown", True, never_hidden=True)]
25
  ]
@@ -32,7 +33,7 @@ for metric in metrics:
32
  auto_eval_column_dict += [
33
  ["model_type", ColumnContent, ColumnContent("Type", "str", False)],
34
  ["architecture", ColumnContent, ColumnContent("Architecture", "str", False)],
35
- ["other_precision", ColumnContent, ColumnContent("Precision", "str", False)], # Overlap with new metric name; might need renaming in context
36
  ["license", ColumnContent, ColumnContent("Hub License", "str", False)],
37
  ["params", ColumnContent, ColumnContent("#Params (B)", "number", False)],
38
  ["likes", ColumnContent, ColumnContent("Hub ❤️", "number", False)],
 
19
  never_hidden: bool = False
20
 
21
  # Dynamic fields based on new data schema
22
+ # metrics = ['precision', 'recall', 'f1_score', 'accuracy', 'yes_percent']
23
+ metrics = ['accuracy']
24
  auto_eval_column_dict = [
25
  ["model_name", ColumnContent, ColumnContent("Model Name", "markdown", True, never_hidden=True)]
26
  ]
 
33
  auto_eval_column_dict += [
34
  ["model_type", ColumnContent, ColumnContent("Type", "str", False)],
35
  ["architecture", ColumnContent, ColumnContent("Architecture", "str", False)],
36
+ ["precision", ColumnContent, ColumnContent("Precision", "str", False)], # Overlap with new metric name; might need renaming in context
37
  ["license", ColumnContent, ColumnContent("Hub License", "str", False)],
38
  ["params", ColumnContent, ColumnContent("#Params (B)", "number", False)],
39
  ["likes", ColumnContent, ColumnContent("Hub ❤️", "number", False)],