ComeçarComece gratuitamente

Alteração do tamanho dos lotes

Você viu que os modelos geralmente são treinados em lotes de tamanho fixo. Quanto menor o tamanho do lote, maior o número de atualizações de peso por época, mas ao custo de uma descida de gradiente mais instável. Especialmente se o tamanho do lote for muito pequeno e não for representativo de todo o conjunto de treinamento.

Vamos ver como diferentes tamanhos de lote afetam a precisão de um modelo de classificação binária simples que separa pontos vermelhos de azuis.

Você usará um tamanho de lote de um, atualizando os pesos uma vez por amostra em seu conjunto de treinamento para cada época. Em seguida, você usará todo o conjunto de dados, atualizando os pesos apenas uma vez por época.

Este exercício faz parte do curso

Introdução à aprendizagem profunda com o Keras

Ver Curso

Exercício interativo prático

Experimente este exercício preenchendo este código de exemplo.

# Get a fresh new model with get_model
model = ____

# Train your model for 5 epochs with a batch size of 1
model.fit(X_train, y_train, epochs=____, ____=____)
print("\n The accuracy when using a batch of size 1 is: ",
      model.evaluate(X_test, y_test)[1])
Editar e executar código