1. Apprendre
  2. /
  3. Cours
  4. /
  5. Introduction aux embeddings avec l'API d'OpenAI

Connected

Exercice

Ajouter des données à la collection

Il est temps d'ajouter ces films et séries télé de Netflix à votre collection ! On vous fournit une liste d'identifiants de documents et de textes, stockés respectivement dans ids et documents, qui ont été extraits de netflix_titles.csv à l'aide du code suivant :

ids = []
documents = []

with open('netflix_titles.csv') as csvfile:
  reader = csv.DictReader(csvfile)
  for i, row in enumerate(reader):
    ids.append(row['show_id'])
    text = f"Title: {row['title']} ({row['type']})\nDescription: {row['description']}\nCategories: {row['listed_in']}"
    documents.append(text)

À titre d'exemple du type d'information qui sera intégré, voici le premier document de documents :

Title: Dick Johnson Is Dead (Movie)
Description: As her father nears the end of his life, filmmaker Kirsten Johnson stages his death in inventive and comical ways to help them both face the inevitable.
Categories: Documentaries

Toutes les fonctions et tous les modules nécessaires ont été importés, et un client persistant a été créé et assigné à client.

Instructions

100 XP
  • Recréez votre collection netflix_titles.
  • Ajoutez les documents et leurs IDs à la collection.
  • Affichez le nombre de documents dans collection et les dix premiers éléments.