我是使用tensorflow的新手,并且对一个热编码的张量流有一些疑问。
我想读一个CVS文件,其中最后一列代表标签。标签是1到7之间的整数值。
我想使用softmax模型进行分类。
因此我需要我的标签采用单张张量格式吗?
是否有一种简单的方法/ tensorflow可以将我的标签转换成热点?
根据文件解析教程,我有以下代码用于解析我的csv,但缺少一个热编码的部分。
"[14,16,18]"
答案 0 :(得分:3)
您可以使用tf.one_hot
# depth = num_clasess in general for the problem
labels = tf.one_hot(tf.stack(attributes[-1]), depth)