keras.utils.to_categorical函数用法

keras.utils.to_categorical函数

问题描述

今天再次精读论文"CoLight: Learning Network-level Cooperation for Traffic Signal Control"的源码，其中邻接矩阵的用法中有这一函数。

to_categorical函数

作用:将原向量变为one-hot编码，用法：

#调用to_categorical将vector按照num_classes个类别来进行转换
l = to_categorical(vector, num_classes)

科普one-hot编码

one-hot编码又称为一位有效编码，主要是采用N位状态寄存器来对N个状态进行编码，每个状态都由他独立的寄存器位，并且在任意时候只有一位有效。

自然编码：000,001,010,011,100,101
one-hot编码：000001,000010,000100,001000,010000,100000

举个栗子

#调用to_categorical将vector按照num_classes个类别来进行转换
l = to_categorical([2,0,1,3], num_classes=6)

结果：
[[0. 0. 1. 0. 0. 0.]
 [1. 0. 0. 0. 0. 0.]
 [0. 1. 0. 0. 0. 0.]
 [0. 0. 0. 1. 0. 0.]]

额外收获：

#axis=-1时表示对按照最后一个轴进行排序
numpy.ndarray.sort(axix=-1)