Keras Fit_generator回調

Question

我正在使用fit_generator從文件讀取數據，當它到達文件末尾時，它會從下一個文件加載數據。我在keras中也使用有狀態的RNN，這樣我需要手動重置狀態，在這種情況下，每當生成器加載一個新文件時。我怎樣才能做到這一點？

發電機看起來是這樣的：

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), ...)