tensorflow可視化Keras框架中Tensorboard使用示例

更新時(shí)間：2022年05月07日 11:47:19 作者：Bubbliiiing

這篇文章主要為大家介紹了tensorflow可視化Keras框架中Tensorboard使用示例，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪

Tensorboard詳解

該類在存放在keras.callbacks模塊中。擁有許多參數(shù)，主要的參數(shù)如下：

1、log_dir: 用來(lái)保存Tensorboard的日志文件等內(nèi)容的位置

2、histogram_freq: 對(duì)于模型中各個(gè)層計(jì)算激活值和模型權(quán)重直方圖的頻率。

3、write_graph: 是否在 TensorBoard 中可視化圖像。

4、write_grads: 是否在 TensorBoard 中可視化梯度值直方圖。

5、batch_size: 用以直方圖計(jì)算的傳入神經(jīng)元網(wǎng)絡(luò)輸入批的大小。

6、write_images: 是否在 TensorBoard中將模型權(quán)重以圖片可視化。

7、update_freq: 常用的三個(gè)值為’batch’ 、 ‘epoch’ 或整數(shù)。當(dāng)使用 ‘batch’ 時(shí)，在每個(gè) batch 之后將損失和評(píng)估值寫入到 TensorBoard 中。 ‘epoch’ 類似。如果使用整數(shù)，會(huì)在每一定個(gè)樣本之后將損失和評(píng)估值寫入到 TensorBoard 中。

默認(rèn)值如下：

log_dir='./logs',  # 默認(rèn)保存在當(dāng)前文件夾下的logs文件夾之下
histogram_freq=0,
batch_size=32,
write_graph=True,  #默認(rèn)是True，默認(rèn)是顯示graph的。
write_grads=False,
write_images=False,
update_freq='epoch'

使用例子

以手寫體為例子，我們打開histogram_freq和write_grads，也就是在Tensorboard中保存權(quán)值直方圖和梯度直方圖。

打開CMD，利用tensorboard --logdir=logs生成tensorboard觀測(cè)網(wǎng)頁(yè)。

1、loss和acc

2、權(quán)值直方圖

3、梯度直方圖

實(shí)現(xiàn)代碼

import numpy as np
from keras.layers import Input, Dense, Dropout, Activation,Conv2D,MaxPool2D,Flatten
from keras.datasets import mnist
from keras.models import Model
from keras.utils import to_categorical
from keras.callbacks import TensorBoard
if __name__=="__main__":
    (x_train,y_train),(x_test,y_test) = mnist.load_data()
    x_train=np.expand_dims(x_train,axis=-1)
    x_test=np.expand_dims(x_test,axis=-1)
    y_train=to_categorical(y_train,num_classes=10)
    y_test=to_categorical(y_test,num_classes=10)
    batch_size=128
    epochs=10
    inputs = Input([28,28,1])
    x = Conv2D(32, (5,5), activation='relu')(inputs)
    x = Conv2D(64, (5,5), activation='relu')(x)   
    x = MaxPool2D(pool_size=(2,2))(x)
    x = Flatten()(x)    
    x = Dense(128, activation='relu')(x)
    x = Dropout(0.5)(x)
    x = Dense(10, activation='softmax')(x)
    model = Model(inputs,x)
    model.compile(loss='categorical_crossentropy', optimizer="adam",metrics=['acc']) 
    Tensorboard= TensorBoard(log_dir="./model", histogram_freq=1,write_grads=True)
    history=model.fit(x_train, y_train, batch_size=batch_size, epochs=epochs, shuffle=True, validation_split=0.2,callbacks=[Tensorboard])

以上就是tensorflow可視化Keras框架中Tensorboard使用示例的詳細(xì)內(nèi)容，更多關(guān)于Keras Tensorboard可視化的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！

您可能感興趣的文章: