PeaceJetのブログ

金融系の社内エンジニアをやりながら、マーケティングやデータ分析などもやっています。

【TensorFlow】TensorBoardでAttributeError

AttributeError: module 'tensorflow.python.training.training' has no attribute 'SummaryWriter'

上記のようなエラーが出た場合、'tensorflow.python.training.training'には 'SummaryWriter'属性は存在しません。
このように言われています。

ドキュメントを見てみると、2016年後半に機能がなくなっており、別のモジュールから呼び出さなければならないそうです。

import tensorflow as tf
sess = tf.Session()
tw = tf.train.SummaryWriter("log", graph=sess.graph)

上記を、下記のように変更してみましょう。
するとうまく行くはずです。

import tensorflow as tf
sess = tf.Session()
tw = tf.summary.FileWriter("log", graph=sess.graph)