a10 commited on
Commit
421aff7
·
1 Parent(s): 4a42d11

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -5
app.py CHANGED
@@ -264,11 +264,18 @@ def plot():
264
 
265
 
266
  for x, y in dataset_val.take(n):
267
- show_plot(
268
- [x[0][:, 1].numpy(), y[0].numpy(), model.predict(x)[0]],
269
- 12,
270
- f"{n} Step Prediction",
271
- )
 
 
 
 
 
 
 
272
 
273
  fig = plot()
274
  st.pyplot(fig)
 
264
 
265
 
266
  for x, y in dataset_val.take(n):
267
+ if ("0" == "mycustom"):
268
+ show_plot(
269
+ [x[0][:, 1].numpy(), y[0].numpy(), model.predict(x)[0]],
270
+ 12,
271
+ f"{n} Step Prediction",
272
+ )
273
+ if ("0" != "mycustom"):
274
+ show_plot(
275
+ [x[0][:, 0].numpy(), y[0].numpy(), model.predict(x)[0]],
276
+ 12,
277
+ f"{n} Step Prediction",
278
+ )
279
 
280
  fig = plot()
281
  st.pyplot(fig)