出现问题:
from tensorflow.contrib.data import Dataset
ImportError: cannot import name 'Dataset'
原因:
tensorflow.contrib.data.Dataset从tensorflow 1.4开始移动为tensor.data.Dataset,成为tensorflow的核心
解决方法:
将代码中
from tensorflow.contrib.data import Dataset
修改为:
from tensorflow.data import Dataset
另外如下问题:
from tensorflow.contrib.data import Iterator
ImportError: cannot import name 'Iterator'
处理方式类似:
from tensorflow.data import Iterator