[英]Keras Fit_generator Callback
我正在使用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
只需将生成器传递给模型:
def gendata(model):
...
model.reset_states()
model.fit_generator(gendata(model), ...)
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.