|
7 | 7 | See LICENSE for the license information |
8 | 8 |
|
9 | 9 | Generate paper-ready plots for the Bayes-tree covariance benchmarks. |
10 | | -Author: Frank Dellaert |
| 10 | +Author: Codex 5.4, prompted by Frank Dellaert |
11 | 11 | """ |
12 | 12 |
|
13 | 13 | import argparse |
@@ -35,9 +35,10 @@ def to_float(rows, fields): |
35 | 35 | row[field] = float(row[field]) |
36 | 36 | row["query_size"] = int(row["query_size"]) |
37 | 37 | row["queries"] = int(row["queries"]) |
38 | | - row["support_cliques"] = int(row["support_cliques"]) |
39 | | - row["compressed_cliques"] = int(row["compressed_cliques"]) |
40 | | - row["reduced_state_dim"] = int(row["reduced_state_dim"]) |
| 38 | + row["repeats"] = int(row["repeats"]) |
| 39 | + row["support_cliques"] = float(row["support_cliques"]) |
| 40 | + row["compressed_cliques"] = float(row["compressed_cliques"]) |
| 41 | + row["reduced_state_dim"] = float(row["reduced_state_dim"]) |
41 | 42 | return rows |
42 | 43 |
|
43 | 44 |
|
@@ -224,7 +225,7 @@ def plot_small_queries(rows, output_path): |
224 | 225 | "w20000\nMETIS", |
225 | 226 | ], |
226 | 227 | ) |
227 | | - ax.set_ylabel("Median query time (ms)") |
| 228 | + ax.set_ylabel("Median per-query mean time (ms)") |
228 | 229 | ax.set_yscale("log") |
229 | 230 | ax.grid(True, axis="y", alpha=0.3) |
230 | 231 | handles, labels = axes[0].get_legend_handles_labels() |
@@ -301,7 +302,7 @@ def plot_ordering(rows, output_path): |
301 | 302 | ax.set_title(dataset_label(dataset)) |
302 | 303 | ax.set_xlabel("Local-window query size") |
303 | 304 | ax.set_xticks(query_sizes, [str(size) for size in query_sizes]) |
304 | | - ax.set_ylabel("Median query time (ms)") |
| 305 | + ax.set_ylabel("Median per-query mean time (ms)") |
305 | 306 | ax.set_yscale("log") |
306 | 307 | ax.grid(True, axis="y", alpha=0.3) |
307 | 308 | handles, labels = axes[0].get_legend_handles_labels() |
@@ -349,7 +350,7 @@ def plot_structure(rows, output_path): |
349 | 350 | ) |
350 | 351 | ax.set_title(dataset_label(dataset)) |
351 | 352 | ax.set_xlabel("Query size") |
352 | | - ax.set_ylabel("Count / dimension") |
| 353 | + ax.set_ylabel("Median count / dimension") |
353 | 354 | ax.grid(True, alpha=0.3) |
354 | 355 | handles, labels = axes[0].get_legend_handles_labels() |
355 | 356 | figure.legend(handles, labels, loc="upper center", ncol=3, frameon=False) |
@@ -767,9 +768,12 @@ def main(): |
767 | 768 | load_rows(input_path), |
768 | 769 | [ |
769 | 770 | "median_total_ms", |
770 | | - "total_total_ms", |
| 771 | + "sum_query_mean_total_ms", |
771 | 772 | "median_reduction_ms", |
772 | 773 | "median_extraction_ms", |
| 774 | + "support_cliques", |
| 775 | + "compressed_cliques", |
| 776 | + "reduced_state_dim", |
773 | 777 | ], |
774 | 778 | ) |
775 | 779 | ) |
|
0 commit comments