如何解決tensorflow添加ptb庫的問題

這篇文章將為大家詳細講解有關如何解決tensorflow添加ptb庫的問題，小編覺得挺實用的，因此分享給大家做個參考，希望大家閱讀完這篇文章后可以有所收獲。

ptb數據集是語言模型學習中應用最廣泛的數據集，常用該數據集訓練RNN神經網絡作為語言預測，tensorflow對于ptb數據集的讀取也定義了自己的函數庫用于讀取，在python 1.0定義了models文件用于導入ptb庫函數，然而當python升級后，導入models文件時就會出現：ModuleNotFountError錯誤，這時需要靠自己下載導入，github上有人共享了models文件，但是不清楚如何安裝，網上教程很多，但是安裝了還有很多的錯誤，本人搗鼓了一天算將其成功導入，因此寫成教程，可以不用下載低版本tensorflow，注意：該教程適用于linux系統下tensorflow。

步驟1：在低版本tensorflow中，導入ptb庫的語句為“from tensorflow.models.rnn.ptb import reader”，其形式與導入mnist庫一樣，因此我們需要查找安裝models庫的位置，在命令行中輸入:

locate tensorflow/examples/tutorials

此時將會顯示出有上面路徑的文件，找到路徑*/tensorflow/examples/tutorials/mnist，此時路徑*/tensorflow就是我們安裝models的路徑，用cd命令進入該文件。

步驟2：進入上面tensorflow文件后，用git下載models文件夾，在命令行中輸入命令：

git clone –recurse-submoduleshttps://github.com/tensorflow/models

如果沒有安裝git，請自行百度如何安裝git

步驟3：此時運行含有語句“from tensorflow.models.rnn.ptb import reader”還是會出錯，主要是因為下載的文件內容與低版本的庫有一定區別，可以逐步進入路徑“*/tensorflow/models”發現，沒有文件rnn，rnn文件存在與路徑“*/tensorflow/models/tutorials/”文件下，因此我們需要將該語句改成

“from tensorflow.models.tutorials.rnn.ptb import reader”

步驟4：此時還會出錯，提示ModuleNotFoundError:No module name ‘reader'，此時我們需要對ptb中的__init__.py文件進行修改，將該文件中的“import reader”修改成“from tensorflow.models.tutorials.rnn.ptb import reader”，還有將“import util”修改成“from tensorflow.models.tutorials.rnn.ptb import util”

此時再次運行程序，將成功導入ptb

關于“如何解決tensorflow添加ptb庫的問題”這篇文章就分享到這里了，希望以上內容可以對大家有一定的幫助，使各位可以學到更多知識，如果覺得文章不錯，請把它分享出去讓更多的人看到。