erhalten Wort von ID bei Tensorflow RNN Probe

Question

Ich versuche, Tensorflow RNN Probe hier zu ändern.

https://www.tensorflow.org/versions/r0.8/tutorials/recurrent/index.html

Bei ptb_word_lm.py denke, ich sie int Array von Wortindex sind Eingeben (m.input_data: x).

Ich möchte tatsächliche Wörter anstelle von IDs sehen, wie kann ich sie sehen?

Answer 1

Sie müssen zunächst das Vokabular (das ein Index von Wort zu ID ist) beibehalten.

An der Spitze der wichtigsten, behalten 4. zurückgegebenen Wert von reader.ptb_raw_data() wie unten.

raw_data = reader.ptb_raw_data(FLAGS.data_path) 
train_data, valid_data, test_data, vocabulary = raw_data 

Dann übergeben Sie das Vokabular zu run_epoch().

test_perplexity = run_epoch(session, mtest, test_data, tf.no_op(), vocabulary) 

Innerhalb des run_epoch(), wenn Sie IDs Worte in dem ersten Schritt von x konvertieren möchten,

def run_epoch(session, m, data, eval_op, vocabulary, verbose=False): 

... 
for step, (x, y) in enumerate(... 

message ="x: " 
for i in range(0, m.num_steps): 
    key = vocabulary.keys()[vocabulary.values().index(x[0][i])] 
    message += key + " " 

print(message) 

Hoffe, es hilft.