1. 程式人生 > >成功解決AttributeError: module 'tensorflow.contrib.data' has no attribute 'TextLineDataset'

成功解決AttributeError: module 'tensorflow.contrib.data' has no attribute 'TextLineDataset'

解決問題

AttributeError: module 'tensorflow.contrib.data' has no attribute 'TextLineDataset'

解決思路

tensorflow版本問題導致的函式呼叫有變更。
(1)、tf.data.TextLineDataset():這個函式的輸入是一個檔案的列表,輸出是一個dataset。dataset中的每一個元素就對應了檔案中的一行。可以使用這個函式來讀入CSV檔案。

解決方法

將舊版本的函式改為tf.data.TextLineDataset()

大功告成!哈哈!