cifar10.load_data()
cifar10.load_data()下载到本地的路径为:
/home/your_userName/.keras/datasets
CIFAR-10数据集官网
https://www.cs.toronto.edu/~kriz/cifar.html
6万张32*32的彩色图片,共10个类,每类6000张图片,其中训练集和测试集的划分如下:
training images: 50000
test images: 10000
训练集被划分为5个batch,每个batch有10000张图片,测试集只有1个batch,也有10000张图片
to_categorical
to_categorical就是将类别向量转换为二进制(只有0和1)的矩阵类型表示。其表现为将原有的类别向量转换为独热编码的形式。先上代码看一下效果:
b = [0,1,8,3]
b = to_categorical(b,9)
print(b)
to_categorical最为keras中提供的一个工具方法,从以上代码运行可以看出,将原来类别向量中的每个值都转换为矩阵里的一个行向量,从左到右依次是0,1,8,3类别。8表示为[0. 0. 0. 0. 0. 0. 0. 0. 1.],只有第8个为1,作为有效位,其余全部为0。即转为one-hot
编码方式
参考博客