参考博客
https://blog.csdn.net/learning_tortosie/article/details/85243310
在本教程中,您将了解Keras .fit
和.fit_generator
函数的工作原理,包括它们之间的差异。为了帮助您获得实践经验,我已经提供了一个完整的示例,向您展示如何从头开始实现Keras数据生成器。
Keras深度学习库包括三个独立的函数,可用于训练您自己的模型:
.fit
.fit_generator
.train_on_batch
-
这三个函数基本上可以完成相同的任务,但他们如何去做这件事是非常不同的。
让我们逐个探索这些函数,查看函数调用的示例,然后讨论它们彼此之间的差异。
调用.fit
:
model.fit(trainX, trainY, batch_size=32, epochs=50)
在这里可以看到提供的训练数据(trainX
)和训练标签(trainY
)。然后,我们指示Keras允许我们的模型训练50
个epoch,同时batch size为32
。
对.fit
的调用在这里做出两个主要假设ÿ