Chọn token kế tiếp theo tham lam
Giải mã tham lam (greedy decoding) luôn chọn token có điểm cao nhất ở mỗi bước. Đây là
Tiến độ của tôi ở bài này
Điểm và code bạn nộp đượ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: python-basics, python-loops.
Nội dung đề bài
Mô tả bài toán
Giải mã tham lam (greedy decoding) luôn chọn token có điểm cao nhất ở mỗi bước. Đây là chiến lược giải mã đơn giản nhất và tất định, làm mốc so sánh cho các chiến lược lấy mẫu phức tạp hơn như top-k hay nucleus.
Yêu cầu
Viết hàm greedy_next_token(logits) trả về chỉ số của token có logit lớn nhất.
Quy ước nộp bài
Nộp hàm greedy_next_token trong solution.py. Hệ thống gọi hàm trực tiếp, so chỉ số với kỳ vọng; không đọc stdin và không in ra stdout.
Input
- logits: dãy điểm số của các token ứng viên.
Output
Một số nguyên là chỉ số của logit lớn nhất. Khi nhiều token bằng điểm, chọn chỉ số nhỏ nhất.
Ràng buộc
- logits phải khác rỗng; ném ValueError nếu rỗng.
- Khi bằng điểm phải chọn chỉ số nhỏ nhất để kết quả tất định.
Ví dụ 1
Input
greedy_next_token(logits=[0.5, 2.0, 1.0])
Output
1
Ví dụ 2
Input
greedy_next_token(logits=[-1.0, -3.0, -2.0])
Output
0
Giải thích
Với logits = [0.5, 2.0, 1.0], giá trị lớn nhất là 2.0 ở chỉ số 1, nên kết quả là 1.
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.
