neural_network/plot.py

import seaborn as sns
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation

sns.set()


def loss_history_plt(loss_history: list) -> None:
    fig, ax = plt.subplots()

    def animate(i):
        ax.clear()
        sns.lineplot(
            x=range(i),
            y=loss_history[:i],
            ax=ax,
        )
        ax.set_xlabel("Epoch")
        ax.set_ylabel("Training Loss")

    _ = FuncAnimation(fig, animate, frames=len(loss_history), interval=100)
    plt.show()