SV08
4 years ago
| 22 | 22 | batch_size = 128 |
| 23 | 23 | maxlen = 125 # 超过这个长度的之后字被丢弃，小于这个长度的在前面补空格 |
| 24 | 24 | plot_every = 20 # 每20个batch 可视化一次 |
| 25 | # use_env = True # 是否使用visodm | |
| 25 | use_env = True # 是否使用visodm | |
| 26 | 26 | env = 'poetry' # visdom env |
| 27 | 27 | max_gen_len = 200 # 生成诗歌最长长度 |
| 28 | 28 | debug_file = '/tmp/debugp' |
| 188 | 188 | |
| 189 | 189 | |
| 190 | 190 | def gen(**kwargs): |
| 191 | # def gen(opt): | |
| 191 | 192 | """ |
| 192 | 193 | 提供命令行接口，用以生成相应的诗 |
| 193 | 194 | """ |
| 194 | 195 | |
| 195 | 196 | for k, v in kwargs.items(): |
| 197 | print(k,v) | |
| 196 | 198 | setattr(opt, k, v) |
| 197 | 199 | data, word2ix, ix2word = get_data(opt) |
| 198 | 200 | model = PoetryModel(len(word2ix), 128, 256); |
| 227 | 229 | |
| 228 | 230 | |
| 229 | 231 | if __name__ == '__main__': |
| 230 | # import fire | |
| 231 | # | |
| 232 | # fire.Fire() | |
| 233 | gen() | |
| 232 | conf={ | |
| 233 | # "max_gen_len":200, # 生成诗歌最长长度 | |
| 234 | # "prefix_words":'漂亮', # 不是诗歌的组成部分，用来控制生成诗歌的意境 | |
| 235 | # "start_words" : '雨' # 诗歌开始 | |
| 236 | } | |
| 237 | result = gen(**conf) | |
| 238 | result = ''.join(result) | |
| 239 | print(result) | |