Schwittleymani / ECO

Electronic Chaos Oracle
https://schwittlick.net/eco
Apache License 2.0
6 stars 1 forks source link

word_level_rnn: fix KeyError #121

Closed schwittlick closed 8 years ago

schwittlick commented 8 years ago
input: dfgdfg
d
e
f
e
c
t
Traceback (most recent call last):
  File "run.py", line 157, in <module>
    result = generator.print_word_rnn_result(input=input_checked)
  File "run.py", line 113, in print_word_rnn_result
    word_level_lstm_author, word_level_lstm_result = generator.sample_word_level_lstm(input=input)
  File "run.py", line 92, in sample_word_level_lstm
    return 'NO AUTHOR YET', selected_word_lstm.sample(input=input, sample=sample, output_length=output_length)
  File "/home/marcel/projects/eco/src/python/word_level_rnn/word_lstm_wrapper.py", line 26, in sample
    return self.model.sample(self.sess, self.words, self.vocab, output_length, input, sample)
  File "/home/marcel/projects/eco/src/python/word_level_rnn/model.py", line 84, in sample
    x[0, 0] = vocab[word]
KeyError: 'defect'
schwittlick commented 8 years ago

happens with this model

peter_sloterdijk-critique_of_cynical_reason