首页 文章

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

提问于
浏览
0

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

我问这个是因为我有一些文章,其中每篇文章都有几个标签 .

1 回答

  • 1

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

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

相关问题