独热编码 One-Hot 编码

独热编码 One-Hot 编码

独热编码即 One-Hot 编码,又称一位有效编码,其方法是使用N位状态寄存器来对N个状态进行编码,每个状态都有它独立的寄存器位,并且在任意时候,其中只有一位有效。

import numpy as np

correct = np.array([0, 1, 2, 3, 4, 5,  6])
n_data = len(correct)

# -- 将 周一至周日 转换为独热编码格式 --
correct_data = np.zeros((n_data, 7))

print(correct_data)
print("\n")

for i in range(n_data):
    correct_data[i, correct[i]] = 1
    
print(correct_data)

例如 周一
[1. 0. 0. 0. 0. 0. 0.]

周二
[0. 1. 0. 0. 0. 0. 0.]

发表回复

您的电子邮箱地址不会被公开。