如何使tensorflow.keras.model.predict输出一个生成器?

问题描述

我知道,# Orders with campaign attribution | order_id | user_id | created_at | campaign | keyword | --------------------------------------------------------- | 123 | abc | 2020-07-04 | facebook | shoes | | 456 | abc | 2020-06-04 | null | null | 可以接收tensorflow.keras.model.predict() generator,但是此函数输出始终是numpy array。有时输出numpy array太大,以至于发生array

当我致电OOM Memory Errors时,是否有任何建议输出发电机?

解决方法

进行发生器包装:

MapItemView

然后在使用时:

MapQuickItems

类似的东西应该可以正常工作。