1. Učit se
  2. /
  3. Kurzy
  4. /
  5. Introduction to Embeddings with the OpenAI API

Connected

cvičení

Řazení podle podobnosti

Teď, když máš všechny příznaky převedené na embeddingy, je dalším krokem výpočet podobností. V tomto cvičení definuješ funkci find_n_closest(), která vypočítá kosinové vzdálenosti mezi dotazovým vektorem a seznamem embeddingů a vrátí n nejmenších vzdáleností spolu s jejich indexy.

V příštím cvičení tuto funkci využiješ k vytvoření aplikace pro sémantické vyhledávání produktů.

distance byl importován z scipy.spatial.

Pokyny

100 XP
  • Vypočítej kosinovou vzdálenost mezi query_vector a embedding.
  • Přidej do seznamu distances slovník obsahující dist a jeho index.
  • Seřaď seznam distances podle klíče 'distance' každého slovníku.
  • Vrať prvních n prvků ze distances_sorted.