Tensorflow多标签分类与部分unlabel

时间:2017-12-31 10:04:17

标签: python tensorflow machine-learning

我想知道张量流是否可以区分负面标签和未知标签。

例如,在新闻分类场景中,我确信该实例属于" sports"和"娱乐",也确定它不属于政治"但不确定它是否属于"游戏"。

如果我使用单热编码处理这些标签,按照" sport","娱乐","政治"和#34;游戏",它应该:

# here I use -1 to represent negative, 0 for unknown
[1, 1, -1, 0]

但是,如果我想用"tf. sigmoid_cross_entropy_with_logits"训练实例,它有两种类型的输出:0或1表示neg和pos标签。

我想要的是"未知标签"不会导致BP的损失,这在张量流中是可能的还是合理的?

非常感谢。

1 个答案:

答案 0 :(得分:0)

我认为你可以将你的三个课程标记为三个一个热门:

正面:(0,0,1)
中性:(0,1,0)
否定:(1,0,0)