keras中to_categorical函数

最新推荐文章于 2024-07-23 16:32:00 发布

weixin_43338538

最新推荐文章于 2024-07-23 16:32:00 发布

阅读量793

点赞数

分类专栏： keras

本文链接：https://blog.csdn.net/weixin_43338538/article/details/89524303

版权

keras 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

to_categorical将类别向量转化为one-hot形式

from keras.utils.np_utils import *

#类别向量
num = [0,1,2,3,4,5,6,7,8]
#调用to_categorical将num按照9个类别来进行转换
num_onehot = to_categorical(num, 9)
print(num_onehot)

结果

[[1. 0. 0. 0. 0. 0. 0. 0. 0.]
 [0. 1. 0. 0. 0. 0. 0. 0. 0.]
 [0. 0. 1. 0. 0. 0. 0. 0. 0.]
 [0. 0. 0. 1. 0. 0. 0. 0. 0.]
 [0. 0. 0. 0. 1. 0. 0. 0. 0.]
 [0. 0. 0. 0. 0. 1. 0. 0. 0.]
 [0. 0. 0. 0. 0. 0. 1. 0. 0.]
 [0. 0. 0. 0. 0. 0. 0. 1. 0.]
 [0. 0. 0. 0. 0. 0. 0. 0. 1.]]