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