Seite 1 von 1

Input_shape eines Dense Layers

Verfasst: Dienstag 5. Juli 2022, 14:44
von nichtSoGuter
Ich arbeite mit dem MNIST Fashion Dataset. Dieses habe ich in X_train, X_test, X_valid aufgeteilt. Dabei hat X_train folgenden shape: (55000, 28, 28) also 55000 instanzen (=bilder), die aus 28x28 pixeln bestehen.

In dem Buch mit dem ich arbeite wird ein Encoder erstellt. Dabei wird X_train.shape[-1] als input_shape eines Dense layers verwendet. X_train.shape[-1] entspricht 28 (da X_train.shape = (55000, 28, 28) ). Der Inputshape müsste doch aber [28,28] sein oder nicht?

Dazu habe ich ein erfundenes Code Beispiel:

Code: Alles auswählen

inputshape = X_train.shape[-1] # entspricht 28
encoder = keras.models.Sequential([
        keras.layers.Dense(10 ,  input_shape=[inputshape]) # müsste input_shape nicht = [28,28] sein?
 ])
Vielen Dank im Voraus!

Re: Input_shape eines Dense Layers

Verfasst: Dienstag 5. Juli 2022, 14:51
von nichtSoGuter
Die frage hat sich geklärt. Im weiteren Verlauf des Codes wird X_train mit X_train_flat = K.batch_flatten(X_train) reshaped. Damit erhält man folgenden shape (-1, 28 * 28).