plot-utils/radar.py at main · FengJin1117/plot-utils · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
# radar_plot.py
import numpy as np
import matplotlib.pyplot as plt


def plot_model_radar(
    model_scores,
    metric_names,
    smaller_is_better=None,
    title=None,
    save_path=None,
):
    """
    Parameters
    ----------
    model_scores : dict
        {
            "ModelA": [m1, m2, ...],
            "ModelB": [m1, m2, ...],
        }
    metric_names : list[str]
        Names of metrics, same order as scores
    smaller_is_better : list[bool]
        Whether each metric is smaller-is-better
    """
    if smaller_is_better is None:
        smaller_is_better = [False] * len(metric_names)

    # --- convert smaller-is-better metrics by reciprocal ---
    processed_scores = {}
    for model, scores in model_scores.items():
        new_scores = []
        for s, sib in zip(scores, smaller_is_better):
            new_scores.append(1.0 / s if sib else s)
        processed_scores[model] = new_scores

    # --- normalize each metric to [0, 1] ---
    score_matrix = np.array(list(processed_scores.values()))
    min_vals = score_matrix.min(axis=0)
    max_vals = score_matrix.max(axis=0)
    norm_scores = (score_matrix - min_vals) / (max_vals - min_vals + 1e-8)

    # --- radar setup ---
    num_metrics = len(metric_names)
    angles = np.linspace(0, 2 * np.pi, num_metrics, endpoint=False).tolist()
    angles += angles[:1]

    fig, ax = plt.subplots(figsize=(6.5, 6.5), subplot_kw=dict(polar=True))

    # angle & grid
    ax.set_theta_offset(np.pi / 2)
    ax.set_theta_direction(-1)
    ax.set_thetagrids(
        np.degrees(angles[:-1]),
        metric_names,
        fontsize=11,
    )

    ax.set_rlabel_position(0)
    ax.set_yticks([0.25, 0.5, 0.75])
    ax.set_yticklabels(["0.25", "0.50", "0.75"], fontsize=9)
    ax.set_ylim(0, 1.0)

    # --- plot each model ---
    for i, (model, values) in enumerate(processed_scores.items()):
        vals = norm_scores[i].tolist()
        vals += vals[:1]

        ax.plot(
            angles,
            vals,
            linewidth=2,
            label=model,
        )
        ax.fill(
            angles,
            vals,
            alpha=0.15,
        )

    if title:
        ax.set_title(title, fontsize=13, pad=16)

    ax.legend(
        loc="upper center",
        bbox_to_anchor=(0.5, -0.12),
        ncol=3,
        frameon=False,
        fontsize=10,
    )

    plt.tight_layout()
    if save_path:
        plt.savefig(save_path, dpi=300)
    plt.show()