2017-01-31 11 views
1

私は以下のコードを使用してデータを訓練しています。しかしここでは、データをバッチに分割する必要があります(MNISTテンソルフローの例:batch_xs、batch_ys = mnist.train.next_batch(100))。分類のためにTensorflowのバッチにデータを分割

next_batchメソッドを検索したとき、どのライブラリでも利用できませんでした。私はテンソルの流れに初心者です。トレーニング中にデータをバッチに分割できるかどうかは不思議です。

本当にありがとうございます。

for i in range(training_epochs): 
    sess.run(optimizer, feed_dict={x: inputX, y_: inputY}) # Take a gradient descent step using our inputs and labels 

# That's all! The rest of the cell just outputs debug messages. 
# Display logs per epoch step 
    if (i) % display_step == 0: 
    cc = sess.run(cost, feed_dict={x: inputX, y_:inputY}) 
    print "Training step:", '%04d' % (i), "cost=", "{:.9f}".format(cc) #, \"W=", sess.run(W), "b=", sess.run(b) 

print "Optimization Finished!" 
training_cost = sess.run(cost, feed_dict={x: inputX, y_: inputY}) 
print "Training cost=", training_cost, "W=", sess.run(W), "b=", sess.run(b), '\n' 

答えて

0

next_batchは、汎用ライブラリメソッドではありません。これは、MNTデータセットに特有の方法であり、次の入力のミニバッチを読み込んで、Tensorflowに送る。

あなたはここでその実装を見ることができます: https://github.com/tensorflow/tensorflow/blob/master/tensorflow/contrib/learn/python/learn/datasets/mnist.py#L160

をあなたのデータセットのための類似した何かを実装する必要があります。 このコードはTensorflowに固有のコードではありません。これを行うPythonライブラリを使用できます。

希望に役立ちます!

関連する問題