AttributeError: module 'tensorflow.contrib.data' has no attribute 'TFRecordDataset' 解决方法

一、报错

当我们使用tf中的dataset时，可能会出现以下报错：

二、原因

tf版本不同导致的错误。

1. 在编写代码时，使用的tf版本可能是TensorFlow 1.3。1.3版本的tf中，Dataset API是放在contrib包中；
2. 但TensorFlow 1.4以后的版本，Dataset API已经从contrib包中移除了，而变成了核心API的一员。

故出现AttributeError: module 'tensorflow.contrib.data' has no attribute 'TFRecordDataset' 报错。

三、解决方法

将

dataset = tf.contrib.data.TFRecordDataset([DATA_PATH])

改为

dataset = tf.data.TFRecordDataset([DATA_PATH])

（即删除contrib.），便能解决上述错误。