在Keras / Tensorflow中,我正在尝试实施一种主动学习机制,我需要为每个样本获得一个损失。我目前在每个样本上执行evaluate
,这需要花费很多时间(每100毫秒一个样本)。有没有一种很好的方法来实现它?
我看到了一个类似的帖子Keras- Loss per sample within batch,但我不明白那里的物流,因为它只是打印。
有没有人用Keras / Tensorflow解决了这个问题?
提前致谢。
编辑:看起来我可以使用fit
为Callback
执行此操作,并将批量大小设为1,但我想为evaluate
或predict
执行此操作