Spaces:
Sleeping
Sleeping
Commit
·
663083f
1
Parent(s):
fbe515e
think that the way we had teh plot function setup may have been causing some problems with buffers in the cloud
Browse files- neural_network/plot.py +4 -4
neural_network/plot.py
CHANGED
@@ -9,13 +9,13 @@ from neural_network.neural_network import NeuralNetwork
|
|
9 |
matplotlib.use("Agg")
|
10 |
|
11 |
def plot(model: NeuralNetwork) -> None:
|
12 |
-
|
|
|
13 |
x=np.arange(len(model.loss_history)),
|
14 |
y=model.loss_history,
|
|
|
15 |
)
|
16 |
buf = io.BytesIO()
|
17 |
-
|
18 |
-
plt.clf()
|
19 |
-
buf.seek(0)
|
20 |
plot_data = base64.b64encode(buf.getvalue()).decode("utf-8")
|
21 |
model.plot = plot_data
|
|
|
9 |
matplotlib.use("Agg")
|
10 |
|
11 |
def plot(model: NeuralNetwork) -> None:
|
12 |
+
fig, ax = plt.subplots()
|
13 |
+
sns.scatterplot(
|
14 |
x=np.arange(len(model.loss_history)),
|
15 |
y=model.loss_history,
|
16 |
+
ax=ax,
|
17 |
)
|
18 |
buf = io.BytesIO()
|
19 |
+
fig.savefig(buf, format="svg")
|
|
|
|
|
20 |
plot_data = base64.b64encode(buf.getvalue()).decode("utf-8")
|
21 |
model.plot = plot_data
|