Madhavan Iyengar
commited on
Commit
·
942f95c
1
Parent(s):
ee55dfa
use only accuracy metric
Browse files- 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 |
-
["
|
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)],
|