generate sequence by indices / one-hot encoding

后端 未结 4 820
时光说笑
时光说笑 2020-12-21 13:36

I have a sequence s = [4,3,1,0,5] and num_classes = 6 and I want to generate a Numpy matrix m of shape (len(s), num_classes)

4条回答
  •  一生所求
    2020-12-21 13:53

    Brocasting still works just do

    $ (labels[:,:,:,None]==np.arange(num_classes))+0

提交回复
热议问题