Тензор потока керас оценивает функцию слишком долго

Question

Я пытался использовать model.evaluate в Google Colab со следующим кодом:

test_datagen = ImageDataGenerator(rescale = 1./255.)
test_generator = test_datagen.flow_from_directory(test_dir,
                                              batch_size = 20,
                                              class_mode = 'binary', 
                                              target_size = (image_size, image_size))
results = model.evaluate(test_generator)
for name, value in zip(model.metrics_names, results):
  print(name, ': ', value)

В консоли я получаю следующий вывод: Найдено 624 изображения, принадлежащих 2 классам. Я ожидал, что модель завершится в ceil (624/20) = 32 шага. Вместо этого функция, кажется, работает вечно (более 1000 шагов). Кто-нибудь знает, что я делаю не так?

Answer 1

В итоге я указал параметр шагов

results = model.evaluate(test_generator, steps=len(test_generator))

len (test_generator) равен 32, как и ожидалось