如何理解keras中的keras.utils.to_categorical

最新推荐文章于 2024-05-22 19:11:57 发布

奶盖芒果

最新推荐文章于 2024-05-22 19:11:57 发布

阅读量5.8k

点赞数 3

分类专栏：深度学习

本文链接：https://blog.csdn.net/candice5566/article/details/105545412

版权

深度学习专栏收录该内容

5 篇文章

订阅专栏

tf.keras.utils.to_categorical(
    y,
    num_classes=None
)

简单来说，keras.utils.to_categorical就是把类别标签转换为onehot编码。

举个例子：

样本标签

1 玫瑰花

2 康乃馨

3 百合花

然后经过one hot encoding，就会转化为：

玫瑰花 100

康乃馨 010

百合花 001

参数：

y: class vector to be converted into a matrix (integers from 0 to num_classes).
num_classes: total number of classes.

A binary matrix representation of the input.

作用：

Converts a class vector (integers) to binary class matrix.

使用：

E.g. for use with categorical_crossentropy.

举个使用的例子：

def preprocess_labels(labels, encoder=None, categorical=True):
    if not encoder:
        encoder = LabelEncoder()
        encoder.fit(labels)
    y = encoder.transform(labels).astype(np.int32)
    if categorical:
        y = np_utils.to_categorical(y)
    return y, encoder