2017-10-14 17 views
1

マルチレイヤパーセプトロンを使用してmnistコードを書きました。しかし、精度と損失関数のスカラーは表示されません(ただし、モデルのグラフは正常に表示されます)。 わかっていれば、私に手がかりを与えることができますか? Tensorflowバージョン:1.2.0テンソルボードにスカラーが表示されない

これはテンソルに表示したい機能です。

def loss(label,y_inf): 
    # Cost Function basic term 
    with tf.name_scope('loss'): 
     cross_entropy = -tf.reduce_sum(label * tf.log(y_inf)) 
    tf.summary.scalar("cross_entropy", cross_entropy) 
    return cross_entropy 



def accuracy(y_inf, labels): 
    with tf.name_scope('accuracy'): 
     correct_prediction = tf.equal(tf.argmax(y_inf, 1), tf.argmax(labels, 1)) 
     accuracy = tf.reduce_mean(tf.cast(correct_prediction, "float")) 
    tf.summary.scalar("accuracy", accuracy) 
    return accuracy 

答えて

1

欠けているかもしれないものは、実際にこれらの要約をフェッチしてディスクに書き込むことです。

まず、あなたがしてFileWriterを定義する必要があります。

fw = tf.summary.FileWriter(LOGS_DIR) # LOGS_DIR should correspond to the path you want to save the summaries in 

次に、単一のOPにすべてのあなたの要約をマージ:あなたが書くことを確認し、あなたのトレーニングのループ内

summaries_op = tf.summary.merge_all() 

今、ディスクへの要約:

for i in range(NUM_ITR): 
    _, summaries_str = sess.run([train_op, summaries_op]) 
    fw.add_summary(summaries_str, global_step=i) 

tensorboardランでこれらの要約を表示するには:

tensorboard --logdir=LOGS_DIR 
関連する問題