如何使用张量流进行多标签分类以达到自动标记的目的?

时间:2018-03-07 09:29:35

标签: tensorflow machine-learning multilabel-classification

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

我之所以这样问是因为我有一系列文章,其中每篇文章都有几个标签。

1 个答案:

答案 0 :(得分:1)

开箱即用,tensorflow通过tf.nn.sigmoid_cross_entropy_with_logits丢失函数等支持二进制多标签分类(请参阅this question中的完整列表)。如果你的标签是二进制的,换句话说,那里有一组预定义的标签,每一个都可以存在,你可以安全地使用它。一个模型可以同时对所有标签进行分类。有很多这样的网络的例子,例如一个来自this question

不幸的是,张量流不支持多名词多标签分类。如果是这种情况,您必须为每个标签构建单独的分类器,每个分类器使用tf.nn.softmax_cross_entropy_with_logits或类似的标签。