загрузить модель и StopIteration

Question

Я использую keras R, keras_model_sequential, fit_generator и generator_pred. Это работает хорошо. И я сохранил модель. Но когда я использую «load_model» и пытаюсь использовать его для «generator_pred», это приводит к этой ошибке.

Произошла ошибка в генераторе: аргумент length.out должен иметь длину 1 Ошибка в py_call_impl (вызываемый, точки $ args, точки $ ключевые слова): StopIteration:

На самом деле единственное отличие состоит в том, что я использую перезагруженную модель, все остальное тоже самое. Кто-то знает почему?

Answer 1

Вы, вероятно, используете генератор "обучения" для "предсказания".

• Тренировочный генератор выводит X и Y.(это означает: длина 2)
• Генератор прогнозирования выводит только X (это означает: длина 1)

Если вы знаете, как получить элементы из генератора в R,Вы можете создать другой генератор, который содержит старый генератор.Новый генератор получает элемент от старого генератора и выводит только первый элемент X, отбрасывая Y.