2017-06-06 6 views
-3

私はGoogleで多くの検索をしましたが、モデルバッチをバッチでトレーニングする例は見つかりませんでした。 Tensorflowのデータセットに組み込まれているmnistにはmnist.train.next_batch(BATCH_SIZE)という関数があります。このような関数が必要です。データセットをバッチに分割し、繰り返しごとに次のバッチをロードします。Tensorflowでモデルバッチをバッチでトレーニングする簡単な例は?

+0

next_batchのソースコードは、無料で入手できます。コードのコードを確実に再利用することができます。 – Wontonimo

答えて

0

あなたは次のように列データを分割することができます:

#Generate a random batch of 200 samples 
rand_index = np.random.choice(len(x_data), size=200) 
rand_x = x_data[rand_index] 
rand_y = y_data[rand_index] 

をし、各反復において、あなたはバッチでバッチを訓練します。

for i in range(1000): 
    for u in range(number_of_batches): 
     sess.run(train, feed_dict={x: rand_x.....}) 

そうであるようなものがあります。

+0

現在、私は同じ方法でトレーニングしていますが、各モデルでは、number_of_batchesとnoを計算する必要があります。反復ごとのバッチ数を手動で計算し、 –