class MySequence(Sequence):
def __init__(self, batchSize): # you can add parameters here
self.batchSize = batchSize
self.xTrain = loadxData() # load your x data here
self.yTrain = loadyData() # load your y data here
def __len__(self):
return self.xData.shape[0]//self.batchSize
def __getitem__(self, index):
return self.xTrain[index*self.batchSize:(index+1)*self.batchSize:]
def on_epoch_end(self):
self.xTrain, self.yTrain = changeData(self.xTrain, self.yTrain) # change your data here
您可以实现一个Sequence,在培训期间为您的模型加载数据。它有一个
on_epoch_end
方法,您可以在下一个epoch开始之前更改数据。在粗略的例子:
然后可以使用^{} 来拟合模型。在
相关问题 更多 >
编程相关推荐