I’m currently learning neuronal networks and trying to understand Tensorflow.
I’ve got a code, where I’m asked about the batch size, but I have no clue.
import tensorflow as tf
mnist = tf.keras.datasets.mnist
(x_train, y_train), (x_test, y_test) = mnist.load_data()
x_train, x_test = x_train / 255.0, x_test / 255.0
model = tf.keras.models.Sequential([
tf.keras.layers.Flatten(input_shape=(28, 28)),
tf.keras.layers.Dense(128, activation='relu'),
tf.keras.layers.Dropout(0.2),
tf.keras.layers.Dense(10, activation='softmax')
])
model.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['accuracy'])
model.fit(x_train, y_train, epochs=5)
I read the documentation and it should be given in the model.fit() method. Default batch size is 32 if I’m right. Is this true for this code?
Thanks in advance
New contributor
tobi is a new contributor to this site. Take care in asking for clarification, commenting, and answering.
Check out our Code of Conduct.