Keras Fit_generator Обратный звонок

Question

Я использую fit_generator, который считывает данные из файла, и когда он добирается до конца файла, он загружает данные из следующего файла. Я также использую RNN с сохранением состояния в keras, так что мне нужно вручную сбросить состояние, в этом случае каждый раз, когда генератор загружает новый файл. Как я могу это достичь?

Генератор выглядит примерно так:

def gendata(): 
    crow = 0 
    path = 'somepath' 
    df = pd.read_csv(path) 
    while True: 
     if x + l < len(df): 
      yield df.iloc[x:x+l,:] 
      x += l 
     else: 
      path = newpath(path) 
      df = pd.read_csv(path) 
      model.reset_states() # this line obviously doesn't work 

Answer 1

просто передать генератор ссылку на модель:

def gendata(model): 
    ... 
    model.reset_states() 

model.fit_generator(gendata(model), ...)