numpy 数组 独热编码 one hot

numpy 数组 独热编码 one hot

import numpy as np 
import tensorflow as tf

array1 = np.array([0, 1, 2, 0, 2])  # 整数分类
print(array1)

[0 1 2 0 2]

one_hot = tf.keras.utils.to_categorical(
    array1,  # 类数组，类值要转换为矩阵（从 0 到 分类数-1 的整数）
    num_classes=3,  # 分类数
    dtype="float32" # 输出数据类型，默认为float32
)
print(one_hot)

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