tf.one_hot用法

- tf.one_hot(

indices,

depth,

on_value=None,

off_value=None,

axis=None,

dtype=None,

name=None)

1. indices是labels，即你要进行one_hot编码的数据

2. depth是你数据的类别，或者区间长度

3. on_value默认是1，这表示对应值有效时，one_hot编码中的值

4. off_value默认是0，表示对应值无效时，one_hot编码的值

5. axis默认为-1，表示你的one_hot编码从哪个方向展开，axis为0时，列表示一个数据，行表示类别

def one_hot_matrix(labels, C):
    """
    Creates a matrix where the i-th row corresponds to the ith class number and the jth column
                     corresponds to the jth training example. So if example j had a label i. Then entry (i,j) 
                     will be 1. 
                     
    Arguments:
    labels -- vector containing the labels 
    C -- number of classes, the depth of the one hot dimension
    
    Returns: 
    one_hot -- one hot matrix
    """

    one = tf.one_hot(indices=labels,depth=C,axis=0)

    sess = tf.compat.v1.Session()
    one_hot = sess.run(one)
    sess.close()
    return one_hot

labels = np.array([1,2,3,0,2,1])
one_hot = one_hot_matrix(labels, C = 4)
print ("one_hot = \n" + str(one_hot))


"""
one_hot = 
[[0. 0. 0. 1. 0. 0.]
 [1. 0. 0. 0. 0. 1.]
 [0. 1. 0. 0. 1. 0.]
 [0. 0. 1. 0. 0. 0.]]
"""

上面的例子，labels表示我要编码的数据，

C表示类别，4表示最终会生成行为4的矩阵

当labels不变，C=5时

labels = np.array([1,2,3,0,2,1])
one_hot = one_hot_matrix(labels, C = 5)
print ("one_hot = \n" + str(one_hot))
"""
one_hot = 
[[0. 0. 0. 1. 0. 0.]
 [1. 0. 0. 0. 0. 1.]
 [0. 1. 0. 0. 1. 0.]
 [0. 0. 1. 0. 0. 0.]
 [0. 0. 0. 0. 0. 0.]]
"""

更多的例子：

下面的例子遵循该函数的设计，更改了传入的参数名称，但在实际使用中上面的例子可能会更加贴近你的想法

indices = [0, 1, 2]
depth = 3
tf.one_hot(indices, depth)  # output: [3 x 3]
# [[1., 0., 0.],
#  [0., 1., 0.],
#  [0., 0., 1.]]

indices = [0, 2, -1, 1]
depth = 3
tf.one_hot(indices, depth,
           on_value=5.0, off_value=0.0,
           axis=-1)  # output: [4 x 3]
# [[5.0, 0.0, 0.0],  # one_hot(0)
#  [0.0, 0.0, 5.0],  # one_hot(2)
#  [0.0, 0.0, 0.0],  # one_hot(-1)
#  [0.0, 5.0, 0.0]]  # one_hot(1)

indices = [[0, 2], [1, -1]]
depth = 3
tf.one_hot(indices, depth,
           on_value=1.0, off_value=0.0,
           axis=-1)  # output: [2 x 2 x 3]
# [[[1.0, 0.0, 0.0],   # one_hot(0)
#   [0.0, 0.0, 1.0]],  # one_hot(2)
#  [[0.0, 1.0, 0.0],   # one_hot(1)
#   [0.0, 0.0, 0.0]]]  # one_hot(-1)

indices = tf.ragged.constant([[0, 1], [2]])
depth = 3
tf.one_hot(indices, depth)  # output: [2 x None x 3]
# [[[1., 0., 0.],
#   [0., 1., 0.]],
#  [[0., 0., 1.]]]