Function plot_grid

tools/static_bench/make_rd_curve.py:229–457

(
    results: OrderedDict,
    metric_keys: List[str],
    styles: Dict[str, Tuple[str, str, str]],
    method_groups: Dict[str, str],
    out_path: Path,
    dpi: int,
    show: bool,
)

def plot_grid(
    results: OrderedDict,
    metric_keys: List[str],
    styles: Dict[str, Tuple[str, str, str]],
    method_groups: Dict[str, str],
    out_path: Path,
    dpi: int,
    show: bool,
) -> None:
    if not metric_keys:
        raise ValueError("At least one metric must be provided.")

    datasets = list(results.keys())
    if not datasets:
        raise ValueError("The input JSON does not contain any dataset entries.")

    n_rows = len(datasets)
    n_cols = len(metric_keys)

    base_height = 3.0
    base_width = base_height * (5.0 / 3.0)
    fig_width = base_width * n_cols
    fig_height = base_height * n_rows

    fig, axes = plt.subplots(
        n_rows,
        n_cols,
        figsize=(fig_width, fig_height),
        squeeze=False,
        gridspec_kw={"wspace": 0.22, "hspace": 0.32},
    )

    methods_with_points: set[str] = set()

    grid_params = {
        "linestyle": "--",
        "alpha": 0.2,
        "color": "gray",
        "which": "both",
        "zorder": 0,
    }

    for row_index, dataset_key in enumerate(datasets):
        dataset_label = format_dataset_label(dataset_key)
        dataset_results = results[dataset_key]

        method_sizes: Dict[str, List[float]] = {}
        method_metrics: Dict[str, Dict[str, List[float]]] = {
            metric_key: {} for metric_key in metric_keys
        }

        for method, variants in dataset_results.items():
            ordered = sorted(variants.values(), key=lambda values: values[3])
            if not ordered:
                continue

            sizes = [values[3] for values in ordered]
            method_sizes[method] = sizes