我想在matplotlib中绘制预测函数与原始函数的关系图。我使用了以下代码:
import matplotlib as plt
plt.ion()
for i in range(200):
#Some Operations
if i%5==0 :
print(str((i+1))+'th loss',loss.squeeze())
# plot and show learning process
plt.cla()
plt.scatter(x.data.numpy(), y.data.numpy())
plt.plot(x.data.numpy(), output.data.numpy(), 'r-', lw=5)
plt.text(0.5, 0, 'Loss=%.4f' % loss.data[0], fontdict={'size': 20, 'color': 'red'})
plt.pause(0.1)
plt.ioff()
plt.show()
执行此代码后,即使我使用cla()
函数清除,结果图也会更新,但它会绘制上一个图。
可以找到整个可执行代码here。