比较与tf.keras.datasets.imdb的功能差异

tf.keras.datasets.imdb

class tf.keras.datasets.imdb()

更多内容详见tf.keras.datasets.imdb。

mindspore.dataset.IMDBDataset

class mindspore.dataset.IMDBDataset(
    dataset_dir,
    usage=None,
    num_samples=None,
    num_parallel_workers=None,
    shuffle=None,
    sampler=None,
    num_shards=None,
    shard_id=None,
    cache=None
)

更多内容详见mindspore.dataset.IMDBDataset。

使用方式

TensorFlow：可使用类内的 load_data 方法下载并加载IMDB数据集。

MindSpore：从指定路径加载IMDB数据集文件，并返回数据集对象。

代码示例

# The following implements IMDBDataset with MindSpore.
import mindspore.dataset as ds

imdb_dataset_dir = "/path/to/imdb_dataset_directory"
dataset = ds.IMDBDataset(dataset_dir=imdb_dataset_dir)

# The following implements imdb with TensorFlow.
import tensorflow as tf

(x_train, y_train), (x_test, y_test) = tf.keras.datasets.imdb.load_data()