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

Question

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

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

Answer 1

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

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