我正在尝试编写一个批处理生成器函数,以便可以使用它训练我的 Keras 模型,但我不知道从哪里开始。有没有人有一些例子或教程?
Keras 生成器功能
数据挖掘
喀拉斯
张量流
2022-02-28 14:37:44
1个回答
本教程以一种简单明了的方式解释了如何创建一个数据生成器,您可以将其传递给您的 Keras 模型以使用fit_generator(). 为了与 Tensorflow 2.x 要求兼容,请记住关于 Keras 生成器的两件事:
- 您的生成器应该继承自
keras.utils.Sequence,这允许训练的内部并行化 - 填充
__getitem__()和__len__()方法。第一个使您的生成器可迭代,第二个使用len生成器对象上的内置命令返回其长度。两者都是 Keras 所必需的。
其它你可能感兴趣的问题