Keras编码解码one-hot标签分类

最新推荐文章于 2024-08-12 08:25:40 发布

zhangphil

最新推荐文章于 2024-08-12 08:25:40 发布

阅读量3k

点赞数

分类专栏： Keras AI 人工智能文章标签： AI Keras Python 人工智能机器学习

本文链接：https://blog.csdn.net/zhangphil/article/details/103277137

版权

AI 同时被 3 个专栏收录

47 篇文章 1 订阅

订阅专栏

人工智能

43 篇文章 0 订阅

订阅专栏

Keras

5 篇文章 1 订阅

订阅专栏

AI：Keras编码解码one-hot标签分类，Python

import keras

if __name__ == "__main__":
    print('\none_hot_1')
    one_hot_1 = keras.utils.to_categorical(np.array([0, 1, 2]))
    print(one_hot_1)

    print('\none_hot_2')
    one_hot_2 = keras.utils.to_categorical(np.array([0, 2, 1]))
    print(one_hot_2)

    print('\none_hot_3')
    one_hot_3 = keras.utils.to_categorical(np.array([2, 1, 0]))
    print(one_hot_3)

    print('\none_hot_4')
    one_hot_4 = keras.utils.to_categorical(np.array([0, 1, 2]), num_classes=4)
    print(one_hot_4)

    print('\none_hot_5')
    one_hot_5 = keras.utils.to_categorical(np.array([0, 2, 1]), num_classes=5)
    print(one_hot_5)

    print('\none_hot_6')
    one_hot_6 = keras.utils.to_categorical(np.array([1, 6]))
    print(one_hot_6)

    print('\none_hot_7')
    one_hot_7 = keras.utils.to_categorical(np.array([6]))
    print(one_hot_7)

输出：

one_hot_1
[[1. 0. 0.]
 [0. 1. 0.]
 [0. 0. 1.]]

one_hot_2
[[1. 0. 0.]
 [0. 0. 1.]
 [0. 1. 0.]]

one_hot_3
[[0. 0. 1.]
 [0. 1. 0.]
 [1. 0. 0.]]

one_hot_4
[[1. 0. 0. 0.]
 [0. 1. 0. 0.]
 [0. 0. 1. 0.]]

one_hot_5
[[1. 0. 0. 0. 0.]
 [0. 0. 1. 0. 0.]
 [0. 1. 0. 0. 0.]]

one_hot_6
[[0. 1. 0. 0. 0. 0. 0.]
 [0. 0. 0. 0. 0. 0. 1.]]

one_hot_7
[[0. 0. 0. 0. 0. 0. 1.]]