keras.utils.to_categorical(y_train, num_classes)
时间: 2024-04-14 10:31:08 浏览: 125
浅谈keras中的keras.utils.to_categorical用法
这是一个使用Keras的utils库中的to_categorical函数,用于将目标变量(例如标签)转换为one-hot编码形式。其中,y_train是训练集的目标变量,num_classes表示类别的数量。这个函数可以将目标变量转换为一个二维矩阵,其中每一行表示一个样本的one-hot编码。每行中的元素对应于相应类别的索引位置,如果该样本属于该类别则为1,否则为0。
阅读全文