在学习word2vec中看到这段程序
if os.path.isfile(save_path):
raise RuntimeError('the save path should be a dir')
if not os.path.exists(save_path):
os.mkdir(save_path)
save_path为输入的保存路径。
意思是输入路径若为文件,提示保存路径应该为目录。输入路径不存在,创建目录。
import os
dirct = 'E:\\study\\文本分类\\程序\\tensorflow-word2vec\\model'
for i in os.listdir(dirct):
print(i)
如上图,运行结果返回5个文件名
checkpoint
params.pkl
tf_vars.data-00000-of-00001
tf_vars.index
tf_vars.meta
我们只需要在使用 os.path.isfile 和 os.path.isdir()时保证输入路径为绝对路径就可以了(使用os.path.join()函数创建,这个函数我在前面有讲os.path.join)