1. 程式人生 > >one-hot中np.eye(n)[x]的意思

one-hot中np.eye(n)[x]的意思

  1. numpy.eye()
    Return a 2-D array with ones on the diagonal and zeros elsewhere.

    Examples:

    >>np.eye(3)
    array([[ 1.,  0.,  0.],
       [ 0.,  1.,  0.],
       [ 0.,  0.,  1.]])
    >>> np.eye(3)[1]
    array([ 0.,  1.,  0.])
    
  2. [label]是陣列元素索引。因此,只有一個元素,它返回給定的行數元素作為陣列。
    Examples:

    >>> np.eye(3)[1]
    array([ 0.,  1.,  0.])
    >>> np.eye(3)[2]
    array([ 0.,  0.,  1.])
    
  3. np.eye(n_labels)[target_vector]
    For example, for a target_vector = np.array([1, 4, 2, 1, 0, 1, 3, 2]), it returns the one-hot coded values:

    np.eye(5)[target_vector]
    Out: 
    array([[ 0.,  1.,  0.,  0.,  0.],
           [ 0.,  0.,  0.,  0.,  1.],
           [ 0.,  0.,  1.,  0.,  0.],
           ..., 
           [ 0.,  1.,  0.,  0.,  0.],
           [ 0.,  0.,  0.,  1.,  0.],
           [ 0.,  0.,  1.,  0.,  0.]])