Spaces:
Sleeping
Sleeping
File size: 569 Bytes
fbe515e 0e83192 b6a95e0 663083f b6a95e0 fbe515e 663083f fbe515e b6a95e0 fbe515e 4629b15 6d15033 fbe515e |
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 |
import numpy as np
import base64
import io
import seaborn as sns
import matplotlib
import matplotlib.pyplot as plt
matplotlib.use("Agg")
def plot(model) -> None:
sns.set()
fig, ax = plt.subplots()
sns.lineplot(
x=np.arange(len(model.loss_history)),
y=model.loss_history,
ax=ax,
)
plt.ylabel("Loss")
plt.xlabel("Epoch")
plt.title("Loss / Epoch")
buf = io.BytesIO()
fig.savefig(buf, format="png")
plt.close(fig)
plot_data = base64.b64encode(buf.getvalue()).decode("utf-8")
model.plot = plot_data
|