Keras DataGenerator:如何返回两个输出,但仅使用一个进行训练?

时间:2019-08-29 11:25:13

标签: python-3.x tensorflow keras

我已经使用keras.utils.Sequence在Keras中定义了一个DataGenerator。我需要获得两个不同的张量作为输出,但是我只希望在训练阶段(y1)中使用其中一个,并使用另一个(y2)作为自定义指标的输入。对于使用同一DataGenerator进行操作,我有一些疑问,但是我找不到其他方法。

我真正需要的是以与输入和y2张量相同的方式在y1中组织数据:以相同的数据顺序进行批处理。我以为我可以在DataGenerator中提供两个张量作为输出,但是据我所知,Keras将生成器返回的所有第二部分都作为训练阶段的输出。

所以我的问题如下:

  1. 我需要以与输入和输出到网络完全相同的方式组织一些数据,以将其用作自定义指标的输入,但在培训阶段完全不使用它:有没有办法不用DataGenerator就能做到吗?

  2. 如果对问题1的回答为“否”,我该如何使用DataGenerator与y2分开工作?我的意思是防止它在训练中产生影响,以后再作为张量使用它。

谢谢。

0 个答案:

没有答案