Tensorflow學(xué)習(xí)筆記3:TensorBoard可視化學(xué)習(xí)
TensorBoard簡(jiǎn)介
Tensorflow發(fā)布包中提供了TensorBoard,用于展示Tensorflow任務(wù)在計(jì)算過(guò)程中的Graph、定量指標(biāo)圖以及附加數(shù)據(jù)。大致的效果如下所示,
TensorBoard工作機(jī)制
TensorBoard 通過(guò)讀取 TensorFlow 的事件文件來(lái)運(yùn)行。TensorFlow 的事件文件包括了你會(huì)在 TensorFlow 運(yùn)行中涉及到的主要數(shù)據(jù)。關(guān)于TensorBoard的詳細(xì)介紹請(qǐng)參考TensorBoard:可視化學(xué)習(xí)。下面做個(gè)簡(jiǎn)單介紹。
Tensorflow的API中提供了一種叫做Summary的操作,用于將Tensorflow計(jì)算過(guò)程的相關(guān)數(shù)據(jù)序列化成字符串Tensor。例如標(biāo)量數(shù)據(jù)的圖表scalar_summary或者梯度權(quán)重的分布histogram_summary。
通過(guò)tf.train.SummaryWriter來(lái)將序列化后的Summary數(shù)據(jù)保存到磁盤(pán)指定目錄(通過(guò)參數(shù)logdir指定)。此外,SummaryWriter構(gòu)造函數(shù)還包含了一個(gè)可選參數(shù)GraphDef,通過(guò)指定該參數(shù),可以在TensorBoard中展示Tensorflow中的Graph(如上圖所示)。
大致的代碼框架如下所示: