问题描述:在利用categorical_crossentropy作为损失函数时,需要将标签设定为one-hot格式,即每个标签的长度应转换为一个长度为类别数的向量,该向量除了所属的类别位置为1之外,其他位置值为0.
from keras.utils.np_utils import to_categorical
categorical_labels = to_categorical(int_labels, num_classes=None)
示例:
import numpy as np
int_labels = np.array([2,1,3,5])
from keras.utils.np_utils import to_categorical
categorical_labels = to_categorical(int_labels, num_classes=None)
print(categorical_labels)

本文介绍如何在Keras中使用to_categorical函数将整数型标签转换为one-hot编码格式,这对于使用categorical_crossentropy损失函数进行训练的神经网络至关重要。通过示例展示,帮助读者理解one-hot编码在深度学习中的应用。

2163

被折叠的 条评论
为什么被折叠?



