Spaces:
Running
Running
Upload folder using huggingface_hub
Browse files- app/draw_diagram.py +2 -2
app/draw_diagram.py
CHANGED
@@ -31,7 +31,7 @@ def draw(folder_name, category_name, dataset_name, metrics):
|
|
31 |
}
|
32 |
|
33 |
data_path = f'{folder}/{category_name.lower()}.csv'
|
34 |
-
chart_data = pd.read_csv(data_path).round(2)
|
35 |
|
36 |
if len(chart_data) == 0:
|
37 |
return
|
@@ -44,7 +44,7 @@ def draw(folder_name, category_name, dataset_name, metrics):
|
|
44 |
dataset_name = dataset_name.replace('-', '_').lower()
|
45 |
chart_data = chart_data[['Model', dataset_name]]
|
46 |
|
47 |
-
chart_data = chart_data.sort_values(by=[dataset_name], ascending=True)
|
48 |
|
49 |
min_value = round(chart_data.iloc[:, 1::].min().min() - 0.1, 1)
|
50 |
max_value = round(chart_data.iloc[:, 1::].max().max() + 0.1, 1)
|
|
|
31 |
}
|
32 |
|
33 |
data_path = f'{folder}/{category_name.lower()}.csv'
|
34 |
+
chart_data = pd.read_csv(data_path).round(2)
|
35 |
|
36 |
if len(chart_data) == 0:
|
37 |
return
|
|
|
44 |
dataset_name = dataset_name.replace('-', '_').lower()
|
45 |
chart_data = chart_data[['Model', dataset_name]]
|
46 |
|
47 |
+
chart_data = chart_data.sort_values(by=[dataset_name], ascending=True).dropna(axis=0)
|
48 |
|
49 |
min_value = round(chart_data.iloc[:, 1::].min().min() - 0.1, 1)
|
50 |
max_value = round(chart_data.iloc[:, 1::].max().max() + 0.1, 1)
|