在使用Keras进行机器学习时,当训练数据较小时,通常使用model.fit()。然而,当训练数据过大时,建议使用model.fit_generator()来替代model.fit()。那么,如何判断数据量已经变得过大呢?
回答:
当你尝试将训练数据加载到内存中时,如果遇到内存错误,就需要切换到fit_generator()。实时生成数据(并从磁盘读取数据)会带来额外的开销,因此在内存中处理数据集进行模型训练总是更快的。
在使用Keras进行机器学习时,当训练数据较小时,通常使用model.fit()。然而,当训练数据过大时,建议使用model.fit_generator()来替代model.fit()。那么,如何判断数据量已经变得过大呢?
回答:
当你尝试将训练数据加载到内存中时,如果遇到内存错误,就需要切换到fit_generator()。实时生成数据(并从磁盘读取数据)会带来额外的开销,因此在内存中处理数据集进行模型训练总是更快的。