Как сбросить генератор данных Keras в R? - PullRequest
1 голос
/ 24 мая 2019

Когда мы имеем дело с генератором данных (для обучения или тестирования) в Keras, бывают случаи, когда мы хотим сбросить генератор.

Следующий код является примером test_generator в R с использованием Keras:

# define the test generator:
test_generator<- flow_images_from_directory(path_to_folder,image_data_generator_object,target_size = c(h_w),class_mode="binary",batch_size =10,shuffle = "FALSE")

#evaluate the model:
my_model%>% evaluate_generator(test_generator, steps = 100)

# Now I want to reset my test_generator before using it for prediction.
# what should be the code?

#predictions
my_model%>%predict_generator(test_generator, steps = 100)

Я хочу, чтобы сбросить генератор данных между этапом оценки и прогнозирования (поскольку порядок не поддерживается.

В Python я обычно так и делаю:

test_generator.reset()

Я не могу найти эквивалент сброса функция в R (керас) . Любая помощь?

1 Ответ

1 голос
/ 24 мая 2019

Вы всегда можете запустить generator$reset().

. Обычно я создаю функцию, которая автоматически запускает генератор.Тогда я могу просто запустить его так:

 my_generator <- function() {
    flow_images_from_data(X_train, Y_train, datagen, batch_size = 32)
  }

  # train using generator
  x <- model %>% fit_generator(my_generator(),
                               steps_per_epoch = 32, 
                               epochs = 2)
...