tensorflow2.0数据集篇(imdb数据集下载)

在tensorflow调用imdb数据集时,经常会出现下载中断或者无法下载的问题

下载的keras中默认的imdb下载地址是
https://s3.amazonaws.com/img-datasets/mnist.npz 但是现在这个地址已经不存在了。
可以将地址改为此网站
https://storage.googleapis.com/tensorflow/tf-keras-datasets/mnist.npz(下载速度很快,其他数据也可以用此网站,修改后面需要的数据集名称即可)

操作如下

在anaconda\envs\tf2\Lib\site-packages\keras\datasets中打开imdb.py文件修改origin=运行文件即可
  if 'nb_words' in kwargs:
        warnings.warn('The `nb_words` argument in `load_data` '
                      'has been renamed `num_words`.')
        num_words = kwargs.pop('nb_words')
    if kwargs:
        raise TypeError('Unrecognized keyword arguments: ' + str(kwargs))

    path = data_utils.get_file(path,
                               origin='https://storage.googleapis.com/tensorflow/tf-keras-datasets/imdb.npz',
                               file_hash='599dadb1135973df5b59232a0e9a887c')
    with np.load(path, allow_pickle=True) as f:
        x_train, labels_train = f['x_train'], f['y_train']
        x_test, labels_test = f['x_test'], f['y_test']

你可能感兴趣的:(imdb数据集,tensorflow,深度学习,机器学习)