时间戳

首页 > 解决方案 > AttributeError：模块“tensorflow_datasets.text”没有属性“YelpPolarityReviews”

python - AttributeError：模块“tensorflow_datasets.text”没有属性“YelpPolarityReviews”

问题描述

我正在尝试下载 TensorFlow 数据集yelp 极性评论

为了做到这一点，我正在执行以下命令：

tfds.text.YelpPolarityReviews

但这会返回以下错误：

AttributeError：模块“tensorflow_datasets.text”没有属性“YelpPolarityReviews”

在此之后，我尝试通过以下方式下载孔数据集：

import tensorflow_datasets.download.DownloadManager as dl_manager
train_dir = dl_manager.download_and_extract('https://s3.amazonaws.com/fast-ai-nlp/yelp_review_polarity_csv.tgz')

但这不起作用，我收到以下错误：

ModuleNotFoundError：没有名为“tensorflow_datasets.download”的模块

问题是如何下载该数据集以便将其与 TensorFlow 一起使用？

谢谢

标签： pythontensorflowkeras

解决方案

数据集是 yelp_polarity_reviews，您必须添加下划线并确保所有内容都在 small 中。你可以试试这个，看看它是否有效？

!pip install tensorflow-datasets
import tensorflow as tf
import tensorflow_datasets as tfds
data = tfds.load('yelp_polarity_reviews', split='train', shuffle_files=True)

推荐阅读