НачатьНачать бесплатно

Эмбеддинг более подробных описаний

Одна из последних предсказанных меток не совсем соответствовала отзыву — скорее всего, это связано с тем, что при эмбеддинге только названий классов модель получает слишком мало информации. На этот раз вместо меток будут использоваться описания каждого класса, чтобы модель лучше «понимала», что именно классифицируется — отзывы о ресторанах.

Для выполнения задания вам доступны следующие объекты:

sentiments = [{'label': 'Positive',
               'description': 'A positive restaurant review'},
              {'label': 'Neutral',
               'description':'A neutral restaurant review'},
              {'label': 'Negative',
               'description': 'A negative restaurant review'}]

reviews = ["The food was delicious!",
           "The service was a bit slow but the food was good",
           "The food was cold, really disappointing!"]

Это упражнение является частью курса

Введение в эмбеддинги с OpenAI API

Посмотреть курс

Инструкции к упражнению

  • Извлеките список описаний тональностей и создайте для них эмбеддинги.

Интерактивное практическое упражнение

Попробуйте выполнить это упражнение, дополнив этот пример кода.

# Extract and embed the descriptions from sentiments
class_descriptions = ____
class_embeddings = ____
review_embeddings = create_embeddings(reviews)

def find_closest(query_vector, embeddings):
  distances = []
  for index, embedding in enumerate(embeddings):
    dist = distance.cosine(query_vector, embedding)
    distances.append({"distance": dist, "index": index})
  return min(distances, key=lambda x: x["distance"])

for index, review in enumerate(reviews):
  closest = find_closest(review_embeddings[index], class_embeddings)
  label = sentiments[closest['index']]['label']
  print(f'"{review}" was classified as {label}')
Редактировать и запускать код