因最近想提高模型的泛化能力,故想知道模型是否过拟合,听实验室的师弟推荐了Tensorboard, 果然好用,记录下tensorboard的使用方法,以防以后踩坑。
keras里调用
from keras.callbacks import TensorBoard
再在model.fit()中插入
callbacks=[TensorBoard(log_dir='mytensorboard')]
远程使用Tensorboard
因我用的服务器跑我的程序,故开始我把网址复制在浏览器里是无法打开网址,查找了原因,是因为无本地访问地址。
解决方法:
1. 登录远程服务器时用
ssh -L 16006:127.0.0.1:6006 account@server.address
2. 到tensorboard文件夹下
cd /usr/local/lib/python2.7/dist-packages/tensorboard
3. 在终端输入
tensorboard --logdir='/home/usrname/' #mytensorboard所在地址
4. 在本地访问地址:
http://127.0.0.1:16006/