2016-11-21 4 views
0

は以下のようにその損失とtrainningを想定します。Tensorflowトレーニングステップで値を取得する方法

cross_entropy = tf.mul(diff, diff) 
train = tf.train.GradientDescentOptimizer(0.1).minimize(cross_entropy) 

は、私は同じように、トレーニングステップにおける重みとバイアスを取得したいと思います:

for i in range(1000): 
    sess.run(train) 
     if cross_entropy == (specific value like 0.1, 0.05): 
      print(weight) 
      print(bias) 

がありますTensorflowでそれを達成する方法はありますか?

答えて

1

はい。

for i in range(1000): 
    _, w_val, b_val, ce_val = sess.run([train, weight, bias, cross_entropy]) 
     if ce_val == 0.005: 
      print(w_val) 
      print(b_val) 
+0

が配列ce_valです:最も簡単な方法は(あなたがあなたのレイヤーからそれらを抽出する必要がない場合、私は、weightbiasはOPSあると仮定)1 run内のすべてのオペレーションを評価し、Pythonでの結果で動作させることです?私はそれを試みましたが、ValueErrorを持っています:複数の要素を持つ配列の真理値はあいまいです。 a.any()またはa.all()を使用する –

+0

ああ、申し訳ありませんが、結果と操作の順序が一致しません。今は大丈夫でしょうか。 – sygi

+0

ありがとうございます。本当に私にとって有益です。 –

関連する問題