分類問題使用 softmax_cross_entropy_with_logits
回歸問題(如預測速度,價格,溫度等)使用均方誤差
loss = tf.reduce_mean(tf.reduce_sum(tf.square(result - prediction), reduction_indices=[1]))
loss = tf.reduce_mean(tf.reduce_sum(tf.square(result - prediction), reduction_indices=[1]))
沒有留言:
張貼留言