Tensorflow中输出向量的二进制掩码

Question

我想在TensorFlow中使用LSTM点击流推荐产品。

我有历史用户行为数据，我想用它来训练模型来推荐产品（在输出上表示为类）但我需要考虑产品在网页上当时是否处于活动状态（不建议不活跃的交易）。

由于我认为使用基本事实非常困难，我想在输出之前使用二进制掩码，然后再与目标矢量进行比较。

在TensorFlow中有没有本地方法可以做到这一点？

Answer 1

您可以在softmax预测输出上使用tf.boolean_mask来删除非活动交易的概率，然后在没有它们的情况下获得最大概率。