手动实现keras.utils.to_categorical

最新推荐文章于 2023-03-10 21:19:16 发布

w112348

最新推荐文章于 2023-03-10 21:19:16 发布

阅读量143

点赞数

分类专栏： python 文章标签：标签编码 one-hot to_categorical 训练标签概率矩阵

本文链接：https://blog.csdn.net/w112348/article/details/114485270

版权

python 专栏收录该内容

7 篇文章 0 订阅

订阅专栏

def to_categorical_1d(train_labels):
    ret = np.zeros((len(train_labels), train_labels.max()+1))
    ret[np.arange(len(ret)), train_labels] = 1
    return ret

a = np.array([5, 0, 4, 1, 9, 2, 1, 3, 1, 4])
R = to_categorical_1d(a)
print(a)
print(R)

[5 0 4 1 9 2 1 3 1 4]

[[0. 0. 0. 0. 0. 1. 0. 0. 0. 0.]
 [1. 0. 0. 0. 0. 0. 0. 0. 0. 0.]
 [0. 0. 0. 0. 1. 0. 0. 0. 0. 0.]
 [0. 1. 0. 0. 0. 0. 0. 0. 0. 0.]
 [0. 0. 0. 0. 0. 0. 0. 0. 0. 1.]
 [0. 0. 1. 0. 0. 0. 0. 0. 0. 0.]
 [0. 1. 0. 0. 0. 0. 0. 0. 0. 0.]
 [0. 0. 0. 1. 0. 0. 0. 0. 0. 0.]
 [0. 1. 0. 0. 0. 0. 0. 0. 0. 0.]
 [0. 0. 0. 0. 1. 0. 0. 0. 0. 0.]]