文本生成的输入字符错误问题

三一王

源代码是以如下的形式进行参数输入的：

1. print(generate_text(model, start_string=u"歌唱祖国："))

复制代码

generate_text函数的开始部分

1. def generate_text(model, start_string):
   
2.   # 评估步骤（用学习过的模型生成文本）
   
3. 
   
4.   # 要生成的字符个数
   
5.   num_generate = 200
   
6. 
   
7.   # 将起始字符串转换为数字（向量化）
   
8.   input_eval = [char2idx[s] for s in start_string]
   
9.   input_eval = tf.expand_dims(input_eval, 0)

复制代码

我想要将输入转为自定义输入，代码如下：

1. txt = input('请给诗词开个头：')
   
2. 
   
3. print(generate_text(model, start_string = txt))

复制代码

input内容为：你好祖国
结果报错：

1. Traceback (most recent call last):
   
2.   File "C:/AI案例/调试/poetry_ada/gettxt.py", line 202, in <module>
   
3.     print(generate_text(model, start_string = txt))
   
4.   File "C:/AI案例/调试/poetry_ada/gettxt.py", line 171, in generate_text
   
5.     input_eval = [char2idx[s] for s in start_string]
   
6.   File "C:/AI案例/调试/poetry_ada/gettxt.py", line 171, in <listcomp>
   
7.     input_eval = [char2idx[s] for s in start_string]
   
8. KeyError: '你'

复制代码

求论坛大佬帮忙看看

suchocolate

字典char2idx没有‘你’这个key，代码不全，卡不出char2idx是从哪里来的。