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