tensorflow - 如何使用 tensorflow 执行多标签分类以实现自动标记？

Question

我是 tensorflow 新手，想知道是否有任何具有多个网络输出的多标签分类的教程或示例。

我问这个问题是因为我有一系列文章，其中每篇文章可以有多个标签。

Answer 1

tensorflow 开箱即用，通过tf.nn.sigmoid_cross_entropy_with_logits支持二进制多标签分类损失函数等(参见 this question 中的完整列表)。如果您的标签是二进制的，换句话说，有一组预定义的可能标签，并且每个标签都可以存在或不存在，那么您可以放心地使用它。单一模型可同时对所有标签进行分类。这种网络有很多例子，例如一份来自 this question .

不幸的是， tensorflow 不支持多项式多标签分类。如果这是你的情况，你必须为每个标签构建一个单独的分类器，每个标签使用 tf.nn.softmax_cross_entropy_with_logits或类似的。