Giải mã vector one-hot về chỉ số lớp
One-hot mã hoá lớp thành vector chỉ có một phần tử 1. Khi mô hình trả về vector như vậy, bước cuối là tìm vị trí của phần tử 1 để biết lớp dự đoán.
Tiến độ của tôi ở bài này
Điểm được lưu vào tài khoản sau khi chấm bài.
Đang tải điểm của bạn…
Kiến thức và chủ đề
Kiến thức tiên quyết: argmax.
Nội dung đề bài
Mô tả bài toán
One-hot mã hoá lớp thành vector chỉ có một phần tử 1. Khi mô hình trả về vector như vậy, bước cuối là tìm vị trí của phần tử 1 để biết lớp dự đoán.
Yêu cầu
Đọc từ stdin: dòng 1 là N; dòng 2 gồm N giá trị 0 hoặc 1, trong đó có đúng một giá trị bằng 1. In ra chỉ số (0-based) của phần tử bằng 1.
Quy ước nộp bài
- Chỉ cần viết một chương trình đọc
stdinvà in rastdout. Bài này không yêu cầu viết hàm. - Không truyền lời nhắc vào
input(). Lời nhắc sẽ bị in vàostdoutvà làm bài sai. - Chỉ in đúng nội dung ở mục Output. Không in thêm nhãn, dòng trống hay ký tự thừa.
- Output được so khớp từng ký tự, phân biệt chữ hoa/thường và dấu câu.
Input
Hai dòng: N và vector one-hot.
Output
Một số nguyên: chỉ số lớp.
Ràng buộc
- 1 ≤ N ≤ 104; vector chứa đúng một giá trị 1.
Ví dụ 1
Input
4
0 1 0 0Output
1
Phần tử 1 nằm ở vị trí 1 (đếm từ 0).Ví dụ 2
Input
3
0 0 1Output
2
Lớp thứ ba nên chỉ số là 2.Gợi ý và lời giải chỉ mở sau khi bạn bấm Nộp bài. Giáo viên và quản trị viên mở được ngay.
Góp ý & báo lỗi bài tập
Đề bài chưa rõ, test có vấn đề hay bạn có ý tưởng giúp bài tốt hơn? Gửi cho đội ngũ AI Empire nhé — mỗi góp ý đều được đọc.
