numpy生成Onehot

编程语言 numpy python

发布时间 : 2020-07-14 00:00

字数:14 阅读 :

import numpy as np

a = [9,2,0]

a_onehot = np.eye(10)[a]


print(a_onehot)

转载请注明来源 https://tianweiye.github.io