Ошибка Tensorflow tf.data.Dataset при использовании функции карты | KeyError

Я работаю над своим замковым проектом. По сути, я пытаюсь создать систему рекомендаций для косметических продуктов Amazon. Набор данных представляет собой набор данных TensorFlow.

Некоторый исходный код, который отлично работает

 data=tfds.load('amazon_us_reviews/Beauty_v1_00', split='train')

 type: tensorflow.python.data.ops.dataset_ops.PrefetchDataset

Отображение некоторой информации о функциях:
```
for sample in data.take(1).as_numpy_iterator():

pprint.pprint(sample)
```

Выход

  {'data': {'customer_id': b'18239070',
       'helpful_votes': 0,
       'marketplace': b'US',
       'product_category': b'Beauty',
       'product_id': b'B00LJ86MAY',
       'product_parent': b'823234087',
       'product_title': b'The Original Curly Tee Towel - T-Shirt Hair Dryi'
                    b'ng Towel Wrap (Extra Long)',
        'review_body': b'Great product, quick ship and packaged nicely with a'
                  b'ttention to detail. Thank you!',
        'review_date': b'2014-10-04',
        'review_headline': b'Very pleased!',
        'review_id': b'R24WHRN0BMM2K7',
        'star_rating': 5,
        'total_votes': 0,
        'verified_purchase': 1,
        'vine': 1}}

Ошибка

Я пытаюсь выбрать только некоторые столбцы, используя функцию карты

       data = data.map(lambda x: {
               "customer_id": x["customer_id"],
               "product_id": x["product_id"],
              "star_rating": x["star_rating"]
              })

KeyError: в пользовательском коде:

       KeyError: 'customer_id'

Код в учебнике работает нормально, но не работает, когда я пытаюсь это сделать. Я гуглил и не мог найти ответ.

Есть ли у вас какие-либо предложения? Теперь спасибо за ваше время.

Кажется, это вызвано тем, как вы получаете доступ к data В учебнике, которому вы следуете, доступ к данным таким образом? Я имею в виду использование: data = data.map(lambda x: { "customer_id": x["customer_id"], "product_id": x["product_id"], "star_rating": x["star_rating"] })

— 17.12.2020 07:41

python tensorflow dictionary tensorflow-datasets keyerror

17.12.2020 07:01

Почему в Python есть оператор "pass"?

Оператор pass в Python - это простая концепция, которую могут быстро освоить даже новички без опыта программирования.

Некоторые методы, о которых вы не знали, что они существуют в Python

Python - самый известный и самый простой в изучении язык в наши дни. Имея широкий спектр применения в области машинного обучения, Data Science,...

Основы Python Часть I

Вы когда-нибудь задумывались, почему в программах на Python вы видите приведенный ниже код?

LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа

Алиса и Боб имеют неориентированный граф из n узлов и трех типов ребер:

Оптимизация кода с помощью тернарного оператора Python

И последнее, что мы хотели бы показать вам, прежде чем двигаться дальше, это

Советы по эффективной веб-разработке с помощью Python

Как веб-разработчик, Python может стать мощным инструментом для создания эффективных и масштабируемых веб-приложений.

686

Перейти к ответу Данный вопрос помечен как решенный

Ответы 1

Ответ принят как подходящий

Вам не хватает ключа «данные» при доступе к словарю.

Это должно исправить это:

data = data.map(lambda x: {
        "customer_id": x["data"]["customer_id"],
        "product_id": x["data"]["product_id"],
        "star_rating": x["data"]["star_rating"]
       })

17.12.2020 11:04