python – 如何使用tf.data.Dataset.from_generator()向生成器
发布时间:2020-12-16 22:28:24 所属栏目:Python 来源:网络整理
导读:我想使用from_generator()函数创建一些tf.data.Dataset.我想向生成器函数(raw_data_gen)发送一个参数.这个想法是生成器函数将根据发送的参数产生不同的数据.通过这种方式,我希望raw_data_gen能够提供培训,验证或测试数据. training_dataset = tf.data.Datase
我想使用from_generator()函数创建一些tf.data.Dataset.我想向生成器函数(raw_data_gen)发送一个参数.这个想法是生成器函数将根据发送的参数产生不同的数据.通过这种方式,我希望raw_data_gen能够提供培训,验证或测试数据.
我尝试以这种方式调用from_generator()时得到的错误消息是:
这是raw_data_gen函数,虽然我不确定你是否需要这个,因为我的预感是问题是调用from_generator():
最佳答案
您需要基于raw_data_gen定义一个不带任何参数的新函数.您可以使用lambda关键字执行此操作.
现在,我们将一个函数传递给from_generator,该函数不带任何参数,但只是将raw_data_gen作为参数设置为1.您可以对验证和测试集使用相同的方案,分别传递2和3. (编辑:李大同) 【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容! |