Несколько запросов и фильтрация
Введение в эмбеддинги с OpenAI API
Emmanuel Pire
Senior Software Engineer, DataCamp
Рекомендации фильмов на основе нескольких точек данных
- Terrifier (id:
's8170') - Strawberry Shortcake: Berry Bitty Adventures (id:
's8103')
Несколько текстов запроса
reference_ids = ['s8170', 's8103']reference_texts = collection.get(ids=reference_ids)["documents"]result = collection.query( query_texts=reference_texts, n_results=3 )
Результат нескольких текстов запроса
{'ids': [['s8170', 's6939', 's7000'],['s8103', 's2968', 's3085']],
'embeddings': None,
'documents': [['Title: Terrifier (Movie)...',
'Title: Haunters: The Art of the Scare (Movie)...',
'Title: Horror Story (Movie)...'],
["Title: Strawberry Shortcake: Berry Bitty Adventures (TV Show)...",
"Title: Shopkins (TV Show)...",
"Title: Rainbow Ruby (TV Show)..."]],
'metadatas': [[None, None, None], [None, None, None]],
'distances': [[0.00, 0.25, 0.26], [0.00, 0.25, 0.28]]}
Добавление метаданных
import csv
ids = []
metadatas = []
with open('netflix_titles.csv') as csvfile:
reader = csv.DictReader(csvfile)
for i, row in enumerate(reader):
ids.append(row['show_id'])
metadatas.append({
"type":row['type'],
"release_year": int(row['release_year'])
})
- Создать список словарей для метаданных
- Создать список идентификаторов для добавления к существующим элементам
Добавление и запрос метаданных
collection.update(ids=ids, metadatas=metadatas)
result = collection.query(
query_texts=reference_texts,
n_results=3,
where={
"type": "Movie"
}
)
Операторы where
where={
"type": "Movie"
}
эквивалентно
where={
"type": {
"$eq": "Movie"
}
}
Список операторов:
$eq— равно (string, int, float)$ne— не равно (string, int, float)$gt— больше (int, float)$gte— больше или равно (int, float)$lt— меньше (int, float)$lte— меньше или равно (int, float)
Несколько фильтров where
where={
"$and": [
{"type":
{"$eq": "Movie"}
},
{"release_year":
{"$gt": 2020}
}
]
}
$or: фильтрация по хотя бы одному условию
Title: A Classic Horror Story (Movie) [...]
===
Title: Nightbooks (Movie) [...]
===
Title: Irul (Movie) [...]
===
Title: Intrusion (Movie) [...]
===
Title: Things Heard & Seen (Movie) [...]
===
Title: A StoryBots Space Adventure (Movie) [...]
Давайте потренируемся!
Введение в эмбеддинги с OpenAI API
