关于训练函数的问题

来源:4-16 综合小练习:梯度下降解决线性回归(下)

金华爱计算机呀

2019-04-26

请问老师,训练方法之前有用过fit(),而现在用train_on_batch()
这两个方法有什么不同么,或者哪个方法训练效果好呢?

写回答

1回答

Oscar

2019-05-06

Keras 的 Sequential 类有三个比较常用且有关联的方法(Sequential 的 英文文档中文文档),可以在文档里搜索这三个方法的定义、参数和使用方法:

  1. fit:适合小型、简单化的数据集。

  2. fit_generator:适合数据集太大而无法放入内存中时。原理是利用生成器(generator),每次载入一个 batch-size 的数据进行训练。比起 train_on_batch 方法,fit_generator 同时可以设置 validation_data,但是采用 train_on_batch  也没关系。

  3. train_on_batch:适合数据集太大而无法放入内存中时。原理是利用生成器(generator),每次载入一个 batch-size 的数据进行训练。

要更深入了解这三个方法的区别和关系,可以参看这篇文章:

如何使用Keras fit和fit_generator(动手教程)

0
1
金华爱计算机呀
非常感谢!
2019-05-09
共1条回复

基于Python玩转人工智能最火框架 TensorFlow应用实践

机器学习入门,打牢TensorFlow框架应用是关键!

2214 学习 · 688 问题

查看课程