Chỉ số của K phần tử lớn nhất

Trung bình

numpysắp xếp4 test · 12 giây · sai số 1e-5

Trả về chỉ số của k phần tử lớn nhất trong x, sắp theo giá trị giảm dần; nếu bằng nhau, chỉ số nhỏ hơn đứng trước. (Dùng khi lấy k dự đoán có xác suất cao nhất — top-k accuracy, top-k sampling.)

Gợi ý: np.argsort(-x, kind='stable')[:k].

Viết hàm

top_k_indices(x, k)

Tham số

x: mảng 1 chiều (n,)
k: 1 ≤ k ≤ n

Trả về

Danh sách k số nguyên.

Ví dụ

Tham số

x = [0.1,0.7,0.2,0.7]
k = 2

Kết quả

[1,3]

Hai giá trị 0.7 → chỉ số 1 trước 3

Tham số

x = [3,1,2]
k = 3

Kết quả

[0,2,1]

Bài tiếp theo: Hàm mất mát MSE →

Python · NumPy

import numpy as np


def top_k_indices(x, k):
    """Chỉ số của K phần tử lớn nhất"""
    # TODO: viết code của bạn ở đây
    pass

import numpy as np


def top_k_indices(x, k):
    """Chỉ số của K phần tử lớn nhất"""
    # TODO: viết code của bạn ở đây
    pass

Bấm “Chạy thử” để gọi top_k_indices() với 2 ví dụ trong đề và so sánh kết quả. Dùng print() thoải mái để gỡ lỗi.