one-hot中np.eye(n)[x]的意思
阿新 • • 發佈:2019-01-13
-
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.])
-
[label]
是陣列元素索引。因此,只有一個元素,它返回給定的行數元素作為陣列。
Examples:>>> np.eye(3)[1] array([ 0., 1., 0.]) >>> np.eye(3)[2] array([ 0., 0., 1.])
-
np.eye(n_labels)[target_vector]
For example, for atarget_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.]])