Mã hóa one-hot
Dễnumpytiền xử lý dữ liệu5 test · 12 giây · sai số 1e-5
Chuyển vector nhãn số nguyên thành ma trận one-hot: hàng i có số 1 ở cột labels[i], còn lại là 0.
- Gợi ý:
np.eye(num_classes)[labels]hoặc gán bằng fancy indexing.
Viết hàm
one_hot(labels, num_classes)Tham số
labels: mảng số nguyên (n,), giá trị trong [0, num_classes)num_classes: số lớp C
Trả về
Ma trận (n, C) gồm 0 và 1.
Ví dụ
Tham số
labels = [0,2,1] num_classes = 3
Kết quả
[[1.0,0.0,0.0],[0.0,0.0,1.0],[0.0,1.0,0.0]]
Tham số
labels = [3,3] num_classes = 5
Kết quả
[[0.0,0.0,0.0,1.0,0.0],[0.0,0.0,0.0,1.0,0.0]]
Bài tiếp theo: Chuẩn hóa z-score theo cột →
Python · NumPy
Bấm “Chạy thử” để gọi one_hot() với 2 ví dụ trong đề và so sánh kết quả. Dùng print() thoải mái để gỡ lỗi.