将标签（int）转换为用于张量流的单热矢量

Question

请帮我解决我的错误。谢谢

这是我的python代码：

Y的形状（199584,1），数据类型为int

num_labels = len(np.unique(Y))

simulated_labels = np.eye(num_labels)[Y]＃一个班轮技巧！

print simulated_labels

错误：

IndexError Traceback（最近一次调用最后一次）  in（）

  1 num_labels = len(np.unique(Y)) # unique labels 681
  2 print num_labels

----> 3 simulated_labels = np.eye(num_labels)[Y] # One liner trick!

  4 print simulated_labels
  5 

IndexError：索引1001超出了轴0的大小为681

的范围

Answer 1

您可以使用tf.one_hot（文档字符串中有示例）