TensorFlow: Wie ist Dataset.train.next_batch definiert?

8

Ich versuche, TensorFlow zu lernen und das Beispiel zu studieren: Ссылка

Ich habe dann einige Fragen im folgenden Code:

%Vor%

Da mnist nur ein Datensatz ist, was genau bedeutet mnist.train.next_batch ? Wie wurde die dataset.train.next_batch definiert?

Danke!

    
Edamame 04.01.2017, 00:40
quelle

1 Antwort

19

Das Objekt mnist wird von read_data_sets() function definiert im Modul tf.contrib.learn . Die Methode mnist.train.next_batch(batch_size) ist implementiert hier und gibt ein Tupel mit zwei Arrays zurück, wobei das erste ein Stapel von batch_size MNIST-Bildern darstellt und das zweite einen Stapel von batch-size -Etiketten, die diesen Bildern entsprechen.

Die Bilder werden als 2-D-NumPy-Array der Größe [batch_size, 784] zurückgegeben (da es 784 Pixel in einem MNIST-Bild gibt), und die Beschriftungen werden entweder als 1-D NumPy-Array der Größe [batch_size] ( wenn read_data_sets() mit one_hot=False aufgerufen wurde) oder ein 2-D NumPy-Array der Größe [batch_size, 10] (wenn read_data_sets() mit one_hot=True aufgerufen wurde).

    
mrry 04.01.2017, 01:07
quelle