one-hot的简单表示 python

乒乒乓乓丫

于 2019-05-17 18:10:59 发布

阅读量462

点赞数

本文链接：https://blog.youkuaiyun.com/qq_39938666/article/details/90298230

版权

深度学习专栏收录该内容

39 篇文章

订阅专栏

import numpy as np

def one_hot(CLASS_NUM, i):
    b = np.zeros(CLASS_NUM)
    b[i] = 1.
    return b

print(one_hot(2,0))
print(one_hot(2,1))
print(*one_hot(2,1))#取出list里的元素
print('--------------------------------')
print(one_hot(10,0))
print(one_hot(10,4))
print(one_hot(10,9))