独热编码即 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.]