TensorFlow 多分类标签转换成One-hot
2017-11-15 21:29
911 查看
TensorFlow 多分类标签转换成One-hot
在处理多分类问题时,将多分类标签转成One-hot编码是一种很常见的手段,以下即为Tensorflow将标签转成One-hot的tensor。以Mnist为例,如果标签为“3”,则One-hot编码为[0,0,0,1,0,0,0,0,0,0].import tensorflow as tf # version : 1.4 NUM_CLASSES = 10 # 10分类 labels = [0,1,2,3] # sample label batch_size = tf.size(labels) # get size of labels : 4 labels = tf.expand_dims(labels, 1) # 增加一个维度 indices = tf.expand_dims(tf.range(0, batch_size,1), 1) #生成索引 concated = tf.concat([indices, labels] , 1) #作为拼接 onehot_labels = tf.sparse_to_dense(concated, tf.stack([batch_size, NUM_CLASSES]), 1.0, 0.0) # 生成one-hot编码的标签
sparse_to_dense 函数说明:https://www.tensorflow.org/api_docs/python/tf/sparse_to_dense
将稀疏矩阵转换成密集矩阵,其中索引在concated中,值为1.其他位置的值为默认值0.
相关文章推荐
- 多分类标签转换为one-hot向量
- PyTorch——Tensor_把索引标签转换成one-hot标签表示
- Tensorflow 的tf.one_hot()功能:dense to one hot
- TensorFlow 之基于Inception V3的多标签分类 retrain
- 【tensorflow】tf.one_hot,tf.reduce_mean,tf.argmax,tf.equal,tf.cast学习
- Keras中制作one-hot标签
- python类别变量(class_label)转换为One_Hot的几种方式
- tensorflow学习笔记--tf.one_hot
- TensorFlow 之基于Inception V3的多标签分类 retrain
- tensorflow--tf.one_hot()函数示例
- Tensorflow学习教程------lenet多标签分类
- 『TensorFlow』one_hot化标签
- CSS资料之标签分类(显示方式)及相应转换
- 【TensorFlow系列】【七】单图片多标签的分类网络搭建与训练
- tensorflow——tf.one_hot以及tf.sparse_to_dense函数
- h5学习之5(html中常见标签的分类和标签之间的转换)
- HTML标签分类:按照标签的显示模式分类
- 21、TensorFlow 实现单层 Softmax 分类
- (四)Tensorflow学习之旅——MNIST分类的卷积神经网络CNN示例
- 机器学习基础 维基翻译 典则相关分析 多标签分类 及简单的sklearn例子