关于训练函数的问题
来源:4-16 综合小练习:梯度下降解决线性回归(下)
金华爱计算机呀
2019-04-26
请问老师,训练方法之前有用过fit(),而现在用train_on_batch()
这两个方法有什么不同么,或者哪个方法训练效果好呢?
写回答
1回答
-
Keras 的 Sequential 类有三个比较常用且有关联的方法(Sequential 的 英文文档 和 中文文档),可以在文档里搜索这三个方法的定义、参数和使用方法:
fit:适合小型、简单化的数据集。
fit_generator:适合数据集太大而无法放入内存中时。原理是利用生成器(generator),每次载入一个 batch-size 的数据进行训练。比起 train_on_batch 方法,fit_generator 同时可以设置 validation_data,但是采用 train_on_batch 也没关系。
train_on_batch:适合数据集太大而无法放入内存中时。原理是利用生成器(generator),每次载入一个 batch-size 的数据进行训练。
要更深入了解这三个方法的区别和关系,可以参看这篇文章:
012019-05-09
相似问题