2017-06-19 7 views
0

私はTensorFlowのGenerative Adversarial Networkに関するチュートリアルに従っています。このチュートリアルでは、MNISTデータセットを使用してモデルをトレーニングします。プログラムの実行速度は向上しますが、使用しているMNISTデータセットのサブセットを取得する方法はわかりません。方法列車セットから長バッチサイズのランダムサンプルを抽出MNISTデータセットをサブセット化する方法は?

mnist.next_batch(batchsize) 

あり

from tensorflow.examples.tutorials.mnist import input_data 
mnist = input_data.read_data_sets("MNIST_data/") 

答えて

0

:以下私はデータセットを抽出するために使用されるコードです。

あなたはランダムな何かをしたくない場合は、テスト・セットのためにmnist.test

x = mnist.train.images[start_batch:end_batch] 
y = mnist.train.labels[start_batch:end_batch] 

または類似してアクセスすることができます。

+0

こんにちは、すばらしく応答してくれてありがとう。私はあなたが提供した方法を使ってtrain.imagesとtrain.labelsをサブセット化することができます。しかし、これらのデータセットのサブセットを作成した後は、NDArrayオブジェクトを取得し、ndarrayのmnistデータセット用に設計されたメソッドを呼び出すことができませんでした。 ndarrayをmnistデータセットにキャストする方法はありますか? – nnguyen24

関連する問題