Keras 生成器功能

数据挖掘 喀拉斯 张量流
2022-02-28 14:37:44

我正在尝试编写一个批处理生成器函数,以便可以使用它训练我的 Keras 模型,但我不知道从哪里开始。有没有人有一些例子或教程?

1个回答

教程以一种简单明了的方式解释了如何创建一个数据生成器,您可以将其传递给您的 Keras 模型以使用fit_generator(). 为了与 Tensorflow 2.x 要求兼容,请记住关于 Keras 生成器的两件事:

  1. 您的生成器应该继承自keras.utils.Sequence,这允许训练的内部并行化
  2. 填充__getitem__()__len__()方法。第一个使您的生成器可迭代,第二个使用len生成器对象上的内置命令返回其长度。两者都是 Keras 所必需的。