John Graham Reynolds commited on
Commit
ca651ac
·
1 Parent(s): ea607d4
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -26,7 +26,7 @@ def populate_map(metric_df: pd.DataFrame, metric_set: set) -> dict:
26
  metric_map = dict()
27
 
28
  for key in metric_set:
29
- for val in metric_df.loc[metric_df["metric"] == key]["average"]:
30
  metric_map[key] = val
31
 
32
  return metric_map
@@ -70,7 +70,7 @@ space = gr.Interface(
70
  label="Table of Predicted vs Actual Class Labels"
71
  ),
72
  gr.Dataframe(
73
- headers=["Metric", "Averaging Type"],
74
  datatype=["str", "str"],
75
  row_count=(3, "fixed"),
76
  col_count=(2, "fixed"),
@@ -83,7 +83,7 @@ space = gr.Interface(
83
  examples=[
84
  [
85
  pd.DataFrame(columns=["Predicted Class Label", "Actual Class Label"], data=[[0,1],[1,1],[2,2],[1,0],[0,0]]),
86
- pd.DataFrame(columns=["Metric", "Averaging Type"], data=[["f1", "weighted"],["precision", "micro"],["recall", "macro"]])
87
  ]
88
  ],
89
  cache_examples=False
 
26
  metric_map = dict()
27
 
28
  for key in metric_set:
29
+ for val in metric_df.loc[metric_df["Metric"] == key]["Averaging Method"]:
30
  metric_map[key] = val
31
 
32
  return metric_map
 
70
  label="Table of Predicted vs Actual Class Labels"
71
  ),
72
  gr.Dataframe(
73
+ headers=["Metric", "Averaging Method"],
74
  datatype=["str", "str"],
75
  row_count=(3, "fixed"),
76
  col_count=(2, "fixed"),
 
83
  examples=[
84
  [
85
  pd.DataFrame(columns=["Predicted Class Label", "Actual Class Label"], data=[[0,1],[1,1],[2,2],[1,0],[0,0]]),
86
+ pd.DataFrame(columns=["Metric", "Averaging Method"], data=[["f1", "weighted"],["precision", "micro"],["recall", "macro"]])
87
  ]
88
  ],
89
  cache_examples=False