Tensorflow学习笔记No.0

这里更新一些学习Tensorflow过程中可能用到的实用工具。

Jupyter Notebook

Jupyter Notebook 是一个非常方便的python编程工具，支持可视化，对于学习python而已非常的实用。

可以使用Anaconda3进行安装。

安装了Tensorflow的小伙伴应该都安装过Anaconda，这里就不再介绍了，正常来说安装Anaconda时会一并安装Jupyter notebook的。

附上基本操作的介绍博客：https://blog.csdn.net/breeze_blows/article/details/84192374

 

matplotlib.pyplot 可视化与绘图

在Jupyter中输入以下代码

import matplotlib.pyplot as plt
%matplotlib inline

成功导入plt模块后即可进行绘图。

以下面代码为例：这是我自己定义的神经网络模型model，训练后使用plt.plot()方法绘制出折线图。

model.compile(optimizer = 'adam',
              loss = 'sparse_categorical_crossentropy',
              metrics = ['acc']
             )

step_per_epochs = train_image.shape[0] // 64

history = model.fit(ds_train, epochs = 5, steps_per_epoch = step_per_epochs,
          validation_data = ds_test, validation_steps = 10000 // 64
         )

plt.plot(history.epoch, history.history.get('acc'), label = 'acc')
plt.plot(history.epoch, history.history.get('val_acc'), label = 'val_acc')
plt.legend()

运行结果：

plt.plot()中，前两个参数分别为图像的x, y轴数据，label是该图线的标签。

plt.legend()是将上面两条图线合并变成一副图像显出来。

这样我们就可以直观的看到训练过程中正确率的变化，判断模型是否欠拟合或者过拟合。

 

Kaggle

　　kaggle是一个比较著名的深度学习的学习与竞赛平台，上面有许多的深度学习数据。

　　可以使用kaggle的notebook来编写程序，并且可以在kaggle提供的GPU上训练网络模型。

　　注册以及使用方法相对简单，自行百度即可。