2017-03-02 47 views
1

我已經通過pip for python 2.7安裝了TensorFlow。閱讀器似乎並未安裝在TensorFlow中

我想運行一些示例RNN模型,其中之一需要讀取器的導入。

from tensorflow.models.rnn.ptb import reader 

-> I get a no module reader found. 

閱讀了一下後,我嘗試了流行的建議:

import reader 

-> no module reader found. 

我跑:

import inspect 
>>> print inspect.getfile(reader) 
Traceback (most recent call last): 
    File "<stdin>", line 1, in <module> 
NameError: name 'reader' is not defined 

是不是讀者應該tensorflow的一部分嗎?我怎樣才能安裝這個模塊?

我需要它,因爲在我試圖運行的代碼,它包括這樣的命令:

def gen_epochs(n, num_steps, batch_size): 
    for i in range(n): 
     yield reader.ptb_iterator(data, batch_size, num_steps) 

回答

0

我不認爲reader.pyhttp://github.com/tensorflow/tensorflow的一部分,但它包含在http://github.com/tensorflow/models中。它在https://github.com/tensorflow/models/tree/master/tutorials/rnn/ptb中使用,並且在3個月前對reader.py提交了一條提交消息,其中提到「將示例模型從github.com/tensorflow/tensorflow移動到github.com/tensorflow/models」。

在教程/ rnn/ptb的示例代碼ptb_word_lm.py中,有一個import reader,我能夠成功運行ptb_word_lm.py

+0

這是有道理的,只是,我運行一個不同的教程,沒有提到(沒有這裏的模型文件夾)。但通過複製它的讀者它的工作。 – dorien