将标签转为onehot编码

最新推荐文章于 2024-01-11 01:55:37 发布

yzffzdong

最新推荐文章于 2024-01-11 01:55:37 发布

阅读量829

点赞数

分类专栏： python 文章标签： python

本文链接：https://blog.csdn.net/weixin_42663286/article/details/111773093

版权

python 专栏收录该内容

5 篇文章 0 订阅

订阅专栏

inp = np.array(['a', 'b', 'c', 'd', 'c', 'b'])
classes = set(inp)            # 元素无重复
# 创建单位矩阵，每一行对应一个one-hot向量，将对应关系用字典存储
classes_dict = {c: np.identity(len(classes))[i, :] for i, c in enumerate(classes)}
# 将每个数据对应的标签表示成one-hot向量
one_hot = np.array(list(map(classes_dict.get, inp)), dtype=np.int32)