TensorFlow下载文件到当前目录

简述

当使用TensorFlow的时候，默认下载文件，如果是windows会进入到C盘的用户界面内。
如果是linux用户，会进入到root目录下。

解决办法

import os
import tensorflow as tf
tf.enable_eager_execution()

train_dataset_url = "http://download.tensorflow.org/data/iris_training.csv"

train_dataset_fp = tf.keras.utils.get_file(fname='./'+os.path.basename(train_dataset_url),
                                           origin=train_dataset_url, cache_dir='.')

print("Local copy of the dataset file: {}".format(train_dataset_fp))

这样就会代码运行的目录下创建一个datasets的目录，下载下来的代码就会放到这里面。

关键点：

• cache_dir='.' 指示目标文件路径。（.表示当前路径）