博客

下述代码源于参考

TensorFlow入门一

中线性回归的实现，想将结果图片保存查看。

1、导入文件：

import matplotlib
matplotlib.use('Agg')
import matplotlib.pyplot as plt

前两行是我添加的，因为我使用的是tensorflow1.0.0版本，运行.py文件中出现错误提示：

File "/usr/lib64/python2.7/lib-tk/Tkinter.py", line 1745, in __init__
    self.tk = _tkinter.create(screenName, baseName, className, interactive, wantobjects, useTk, sync, use)
_tkinter.TclError: no display name and no $DISPLAY environment variable

添加上这两行后错误消失；

2、准备绘图：

因为第一次使用python绘图，我先是写下以下语句想画个500*500的窗口：

plt.figure(figsize=(500, 500)) 

接下来绘图保存：

plt.plot(train_X, train_Y, 'ro', label='Original data')
plt.plot(train_X, sess.run(W) * train_X + sess.run(b), label='Fitted line')
plt.savefig("examples.jpg")

但是执行plt.savefig时遇到段错误，将

plt.figure(figsize=(500, 500)) 

该句删除后，成功绘图；

but，我还不知道why，希望后续可以理解。。。