Невозможно получить plot.ion () в операторе if для оценки прогресса модели

Question

У меня проблема с живым графиком.Я управляю нейронной сетью с тензорным потоком и хочу видеть прогресс через живой график.Код ниже - это то, что у меня сейчас есть.

Это настраивает сюжет

plt.ion()
fig = plt.figure()
ax1 = fig.add_subplot(111)
line1, = ax1.plot(y_test)
line2, = ax1.plot(y_test*0.5)
plt.show()

, тогда я хочу проверить прогресс

if np.mod(i, 5) == 0:
   pred = net.run(out, feed_dict={X: X_test})
   line2.set_ydata(pred)
   plt.pause(0.01)

Я сейчас нахожусьполучает пустой график и не может понять, почему.Пожалуйста, помогите.

Answer 1

plt.ion()
fig = plt.figure()
ax1 = fig.add_subplot(111)
ax1.plot(y_test)
ax1.plot(y_test*0.5)
plt.show()

# Presumably there should be a for loop here
for i in range(max_iter):
    # Code to fit the neural net, "net" to latest batch of data
    # ...

    if i % 5 == 0:
       pred = net.run(out, feed_dict={X: X_test})
       ax1.clear()
       ax1.plot(y_test)
       ax1.plot(pred)
       plt.pause(0.01)