neural_network/plot.py

import numpy as np
import base64
import io
import seaborn as sns
import matplotlib
import matplotlib.pyplot as plt
from neural_network.neural_network import NeuralNetwork

matplotlib.use("Agg")

def plot(model: NeuralNetwork) -> None:
    _ = sns.scatterplot(
        x=np.arange(len(model.loss_history)),
        y=model.loss_history,
    )
    buf = io.BytesIO() 
    plt.savefig(buf, format="svg")
    plt.clf()
    buf.seek(0)
    plot_data = base64.b64encode(buf.getvalue()).decode("utf-8")
    model.plot = plot_data