通过数据生成器解决大文件无法一次性载入内存训练的问题
from keras.models import Model
def process_line(line):
tmp = [float(val) for val in line.strip().split(','
本文介绍如何利用Python生成器解决大文件在训练过程中无法一次性载入内存的问题。通过open()函数获取文件句柄，配合while循环实现多轮读取。Keras的fit_generator函数可以接受生成器提供的数据，进行批量训练。为了提高训练效果，建议在生成数据前进行数据打乱。
最低0.47元/天 解锁文章
扫一扫
4589
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
