Terry Zhuo
commited on
Commit
•
5fa61d0
1
Parent(s):
497f0f1
fix code format
Browse files- src/tools/plots.py +7 -1
src/tools/plots.py
CHANGED
@@ -22,7 +22,13 @@ def plot_solve_rate(df, task, rows=30, cols=38):
|
|
22 |
values = df["solve_rate"]
|
23 |
|
24 |
values = np.array(values, dtype=float) # Ensure values are floats
|
25 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
26 |
n = len(values)
|
27 |
pad_width = rows * cols - n
|
28 |
|
|
|
22 |
values = df["solve_rate"]
|
23 |
|
24 |
values = np.array(values, dtype=float) # Ensure values are floats
|
25 |
+
|
26 |
+
# Extract numerical IDs and sort by them
|
27 |
+
ids = [int(key.split('/')[-1]) for key in keys]
|
28 |
+
sorted_indices = np.argsort(ids)
|
29 |
+
keys = np.array(keys)[sorted_indices]
|
30 |
+
values = values[sorted_indices]
|
31 |
+
|
32 |
n = len(values)
|
33 |
pad_width = rows * cols - n
|
34 |
|