python怎么把one-hot转化为一维标签

时间:2024-10-12 09:11:55

1、如下图所示,假设我们总共有4类物品,标签分别为"0-3"。

python怎么把one-hot转化为一维标签

2、我们先使用“TensorFlow”将标签转化为“one-hot”的形式。

python怎么把one-hot转化为一维标签

3、但是“one-hot”向原始的一维标签转化的话,就没有对应的函数了~不过理清思锭缠愁颊路后也十分简单。首先我们需要遍历每一个“one-hot”。

python怎么把one-hot转化为一维标签

4、接下来我们针对每一个“one-hot”要找回对应的“标签”:可以发现,“one-hot”编码中“1”是特殊标记,我们只需要获取“1”的位置即可,这一点的实现numpy为我们提供了遍历!

python怎么把one-hot转化为一维标签

5、接下来就是使用“列表表达式”压缩代码啦~

python怎么把one-hot转化为一维标签
© 手抄报圈