1. Learn
  2. /
  3. Courses
  4. /
  5. Nhập môn Embeddings với OpenAI API

Connected

Exercise

Sắp xếp theo độ tương đồng

Giờ bạn đã nhúng xong tất cả đặc trưng, bước tiếp theo là tính độ tương đồng. Trong bài tập này, bạn sẽ định nghĩa hàm find_n_closest(), hàm này tính khoảng cách cosine giữa một vector truy vấn và danh sách các embedding, rồi trả về n khoảng cách nhỏ nhất kèm theo chỉ số của chúng.

Ở bài tập kế tiếp, bạn sẽ dùng hàm này để xây dựng ứng dụng tìm kiếm ngữ nghĩa cho sản phẩm.

distance đã được nhập từ scipy.spatial.

Instructions

100 XP
  • Tính khoảng cách cosine giữa query_vector và embedding.
  • Thêm vào danh sách distances một dictionary chứa dist và index của nó.
  • Sắp xếp danh sách distances theo khóa 'distance' của mỗi dictionary.
  • Trả về n phần tử đầu tiên trong distances_sorted.