summaryrefslogtreecommitdiffstats
path: root/training/lr.py
blob: f5b362f3e69c2e6d2eb42ea3c42f7dea7e36b4c9 (plain) (blame)
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
from pathlib import Path

import matplotlib.pyplot as plt


def plot_metrics(
    metrics: tuple[list[float], list[float], list[float]],
    output_file: Path,
    skip_start: int = 10,
    skip_end: int = 5,
):
    lrs, losses, accs = metrics

    if skip_end == 0:
        lrs = lrs[skip_start:]
        losses = losses[skip_start:]
        accs = accs[skip_start:]
    else:
        lrs = lrs[skip_start:-skip_end]
        losses = losses[skip_start:-skip_end]
        accs = accs[skip_start:-skip_end]

    fig, ax_loss = plt.subplots()
    ax_acc = ax_loss.twinx()

    ax_loss.plot(lrs, losses, color='red')
    ax_loss.set_xscale("log")
    ax_loss.set_xlabel(f"Learning rate")
    ax_loss.set_ylabel("Loss")

    ax_acc.plot(lrs, accs, color='blue')
    ax_acc.set_xscale("log")
    ax_acc.set_ylabel("Accuracy")

    plt.savefig(output_file, dpi=300)
    plt.close()