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