1、如下图所示,假设我们总共有4类物品,标签分别为"0-3"。
2、我们先使用“TensorFlow”将标签转化为“one-hot”的形式。
3、但是“one-hot”向原始的一维标签转化的话,就没有对应的函数了~不过理清思锭缠愁颊路后也十分简单。首先我们需要遍历每一个“one-hot”。
4、接下来我们针对每一个“one-hot”要找回对应的“标签”:可以发现,“one-hot”编码中“1”是特殊标记,我们只需要获取“1”的位置即可,这一点的实现numpy为我们提供了遍历!
5、接下来就是使用“列表表达式”压缩代码啦~