【问题标题】：Convert labels(int) into one-hot vectors for tensorflow将标签（int）转换为张量流的单热向量
【发布时间】：2017-05-28 09:28:50
【问题描述】：

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

这是我的python代码：

Y 的形状 (199584, 1) 和数据类型是 int

num_labels = len(np.unique(Y))

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

打印模拟标签

错误：

IndexError Traceback（最近一次调用最后一次）在 ()

  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

【问题讨论】：

如果 Y 中缺少标签，它们将不会被 len(np.unique(Y)) 计算在内，请尝试 np.max(Y)+1。
我回答你的问题了吗？

标签： tensorflow

【解决方案1】：

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

【讨论】：