python - 用于批量训练的张量流中的一种热编码

Question

我的训练数据包含约 1500 个标签（字符串，每条记录一个标签），我想做批量训练（只需将一批加载到内存中以更新神经网络中的权重）。我想知道 tensorflow 中是否有一个类可以对每批中的标签进行一次热编码？我们可以在sklearn中做一些事情

onehot_encoder = OneHotEncoder(sparse=False)
onehot_encoder.fit(entire training labels)

然后在 tensorflow 会话中的每个批次中，我可以转换我的批次标签并输入 tensorflow 进行训练

batch_label = onehot_encoder.transform(batch training labels)
sess.run(feed_dict={x:...,y:batch_label)

一个例子将不胜感激。谢谢。

score 0 · Accepted Answer

我认为这篇文章类似于那个Tensorflow One Hot Encoder？

indices = [0, 1, 2]
depth = 3
tf.one_hot(indices, depth)  
# output: [3 x 3]
# [[1., 0., 0.],
#  [0., 1., 0.],
#  [0., 0., 1.]]

只是发布它以节省您的时间；）

1 回答 1