tensorflow 学习笔记5 结果可视化

tensorflow结果可视化需要安装matplotlib库

import matplotlib.pyplot as plt

然后进行可视化

#可视化
#作用新建绘画窗口,独立显示绘画的图片
fig=plt.figure()
#画布为一行一列一块
ax=fig.add_subplot(1,1,1)
#输入的数据为x_data y_data
ax.scatter(x_data,y_data)
#show以后不暂停
plt.ion()
#这个函数是显示绘画的图像
plt.show()

显示可视化结果：

        try:
            #新增一条线之后去除原来的线
            ax.lines.remove(lines[0])
        except Exception:
            pass
        print(sess.run(loss,feed_dict={xs:x_data,ys:y_data}))
        prediction_value=sess.run(prediction,feed_dict={xs:x_data, ys: y_data})
        #可视化的连续线：x值，y值，红色，宽度5
        lines=ax.plot(x_data,prediction_value,'r-',lw=5)
        #每循环一次暂停0.1秒
        plt.pause(0.1)

结果图片：