【TensorFlow学习笔记】完美解决 pip3 install tensorflow 没有models库，读取PTB数据

安装tensorflow

我使用的是最最最简单的容易的 pip3 install

《TensorFlow学习笔记》一、安装win10下python3.6的tensorflow的CPU版本安装

读取PTB数据集发现问题

之前一直用的好好的，最近研究RNN 和LSTM 需要用到PTB数据集，这个数据集在介绍的用法里可以直接调用tensorflow的包进行read 出训练集，测试集，验证集 PTB 可以直接百度下载
PTB目录为
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-EqD6S3E5-1570588720426)(https://img-blog.csdn.net/20180224123138311?watermark/2/text/aHR0cDovL2Jsb2cuY3Nkbi5uZXQvcXFfMzM0MzEzNjg=/font/5a6L5L2T/fontsize/400/fill/I0JBQkFCMA==/dissolve/70)]
需要的数据集在data下分别为ptb.test.txt ptb.train.txt ptb.valid.txt

代码如下

from tensorflow.models.rnn.ptb import reader
DATAPATH = './simple-example/data'
train_data, valid_data, test_data, _ = reader.ptb_raw_data(DATAPATH)

问题

发现pip3方式下安装的tensorflow没有models库
上网上查了一下得知
目前tf1.0版本的也就是比较新的版本都将models库剔除，当做第三方库来使用了
解决方案：

直接重新下载安装 tf的低的版本
手动安装models库

安装models库

我首先尝试直接下载github上的models
直接放到tf文件夹下，后来发现tf根本找不到这个包，本人不才（因为平时用包全都是直接pip3 install的方式所以没太在意这些东西），主要原因不知为何，我猜是因为此刻这个的models相当于第三方库，这里直接复制到文件夹下没有编译，所以根本没用
查了下这个models库的用法可以通过git命令进行clone到tensorflow文件下
① 安装git 这个百度多的是就不多说了
② 手动找到python36下的文件tensorflow文件夹

进行鼠标右键点击选择 Git Bash here
（这个过程也可以先在任何一个地方Git Bash here 之后利用cd 到tf文件夹里也一样）
③
在命令框里输入

git clone --recurse-submodules https://github.com/tensorflow/models

安装编译完毕
这回你会发现tf文件夹下已经有models文件夹了

导入程序试试看

from tensorflow.models.tutorials.rnn.ptb import reader#存在models了也可以导入了

随便写个整体程序

from tensorflow.models.tutorials.rnn.ptb import readerDATAPATH = './simple-examples/data'
train_data, valid_data, test_data, _ = reader.ptb_raw_data(DATAPATH)print(len(train_data))print(train_data[:100])

万万没想到又出现一个问题

说找不到reader 这个
后来发现问题出在ptb的__init__.py文件下
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-cx5Rp5Qd-1570588720432)(https://img-blog.csdn.net/20180224125308789?watermark/2/text/aHR0cDovL2Jsb2cuY3Nkbi5uZXQvcXFfMzM0MzEzNjg=/font/5a6L5L2T/fontsize/400/fill/I0JBQkFCMA==/dissolve/70)]
发现这个import 是错的没法import 进去reader
我又查看了一下其他的第三方库的这个初始文件发现需要这么操作就解决了

其实问题就是出在这个导入其他的py文件的时候导入出错了。

from tensorflow.models.tutorials.rnn.ptb import reader
from tensorflow.models.tutorials.rnn.ptb import util

经过以上的这么折腾可算全部KO，这回再走一遍那个读取PTB数据的程序

from tensorflow.models.tutorials.rnn.ptb import readerDATAPATH = './simple-examples/data'
train_data, valid_data, test_data, _ = reader.ptb_raw_data(DATAPATH)print(len(train_data))print(train_data[:100])

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-LnwUmih0-1570588720434)(https://img-blog.csdn.net/20180224125612961?watermark/2/text/aHR0cDovL2Jsb2cuY3Nkbi5uZXQvcXFfMzM0MzEzNjg=/font/5a6L5L2T/fontsize/400/fill/I0JBQkFCMA==/dissolve/70)]

完美运行

PS：如果觉得本篇本章对您有所帮助，欢迎关注、评论、点赞！