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)
|