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

Question

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

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

由于我认为使用地面实况非常困难，我想在输出与目标向量进行比较之前使用二进制掩码。

在 TensorFlow 中是否有任何本地方法可以做到这一点？

Answer 1

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