首页 > 其他 > 详细

one_hot编码

时间:2019-08-12 11:20:35      阅读:83      评论:0      收藏:0      [点我收藏+]

one_hot参数说明

 1 In[]:tf.one_hot([[13, 25, 14, 15], [19, 23, 20, 16]], depth=26, axis=2, on_value=1.0).eval()
 2 
 3 Out[]:array([[[0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0.,
 4          0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],
 5         [0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,
 6          0., 0., 0., 0., 0., 0., 0., 0., 0., 1.],
 7         [0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0.,
 8          0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],
 9         [0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1.,
10          0., 0., 0., 0., 0., 0., 0., 0., 0., 0.]],
11        [[0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,
12          0., 0., 0., 1., 0., 0., 0., 0., 0., 0.],
13         [0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,
14          0., 0., 0., 0., 0., 0., 0., 1., 0., 0.],
15         [0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,
16          0., 0., 0., 0., 1., 0., 0., 0., 0., 0.],
17         [0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,
18          1., 0., 0., 0., 0., 0., 0., 0., 0., 0.]]], dtype=float32)

将一个形状(2, 1, 4)的array转换成one_hot编码,

depth表示最大特征值数量,

axis=2表示按形状中第三个数值‘4列‘进行转换,

on_value=1.0表示转换成浮点类型字符。

one_hot编码

原文:https://www.cnblogs.com/yanhai1210/p/11337928.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!