keras中使用tensorboard

1.在keras搭建的神经网络中使用tensorboard

主要看需要导入tensorboard,并在model.fit()的callbacks中设置参数,很多代码都是根据epoch的次数来绘制loss值和acc值的,想根据batch(即迭代次数绘制loss值和acc值,只需要更改update_freq的参数为"batch"),tensboard其他常用参数可参考 神经网络学习小记录43——Keras中Tensorboard的使用,我的设置如下:

from keras.callbacks import TensorBoard
model.fit(X_train, Y_train,
            batch_size=batch_size, nb_epoch=nb_epoch,
            verbose=1,
            validation_data=(X_test, Y_test),
            callbacks=[history,model_checkpoint,TensorBoard(log_dir=r"./mytensotboard",
                        histogram_freq=1,update_freq="batch",)])

根据迭代次数绘制的loss值变化如下图:
在这里插入图片描述

2.tensorboard在游览器中显示

最好使用Google游览器,其他游览器可能不能显示,在cmd窗口输入:tensorboard --logdir 保存的log路径 --host=127.0.0.1,我在cmd窗口输入如下

tensorboard --logdir E:\jupyter_notebook\U_net_rural_road_extraction\mytensotboard --host=127.0.0.1

最后附上使用minst数据集实验的全部代码,其中losshistory类似乎只能根据epoch记录loss和acc的值,可以去掉:

from __future__ import print_function
import numpy as np
np.random.seed(1337)  # for reproducibility

import keras
from keras.datasets import mnist
from keras.models import Sequential
from keras.layers.core import Dense, Dropout, Activation
from keras.optimizers import SGD, Adam, RMSprop
from keras.utils import np_utils
import matplotlib.pyplot as plt
from keras.callbacks import ModelCheckpoint
from keras.callbacks import TensorBoard
# %matplotlib inline

#写一个LossHistory类,保存loss和acc
class LossHistory(keras.callbacks.Callback):
    def on_train_begin(self, logs={
   }):
        self.losses = {
   'batch':[], 'epoch'
  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值