56ccfee
SV08 4 years ago
2 changed file(s) with 32 addition(s) and 39 deletion(s). Raw diff Collapse all Expand all
+21
-7
handler.py less more
88 """
99
1010 # param1 = conf['param1'] # value_type: str # description: some description
11 start_words = conf['start_words'] #诗歌开始
12 prefix_words = conf['prefix_words'] #诗歌语境
13 max_gen_len = conf['max_gen_len'] #诗歌最大长度
14 # add your code
11 # start_words = conf['start_words'] #诗歌开始
12 # prefix_words = conf['prefix_words'] #诗歌语境
13 # max_gen_len = conf['max_gen_len'] #诗歌最大长度
14 # # add your code
15 # if __name__ == '__main__':
16 # cnf = {start_words,prefix_words,max_gen_len}
17 # result = main.gen(**cnf)
18 # result = ''.json(result)
19 # return {'result': result}
20
1521 if __name__ == '__main__':
16 result = main.gen(start_words,prefix_words,max_gen_len)
17 result = ''.json(result)
18 return {'result': result}
22 start_words = conf['start_words'] #诗歌开始
23 prefix_words = conf['prefix_words'] #诗歌语境
24 max_gen_len = conf['max_gen_len'] #诗歌最大长度
25 cof={
26 "max_gen_len": max_gen_len, # 生成诗歌最长长度
27 "prefix_words":prefix_words, # 不是诗歌的组成部分，用来控制生成诗歌的意境
28 "start_words" : start_words # 诗歌开始
29 }
30 result = main.gen(**cof)
31 result = ''.join(result)
32 return {'ret1':result}
1933 ⏎
+11
-32
main.py less more
3232 acrostic = False # 是否是藏头诗
3333 model_prefix = 'checkpoints/tang' # 模型保存路径
3434
35 # conf={
36 # "data_path":'tang.npz', # 诗歌的文本文件存放路径
37 # "pickle_path":'tang.npz' , # 预处理好的二进制文件
38 # "author":None, # 只学习某位作者的诗歌
39 # "constrain": None, # 长度限制
40 # "category" :'poet.tang', # 类别，唐诗还是宋诗歌(poet.song)
41 # "lr" :1e-3,
42 # "weight_decay": 1e-4,
43 # "use_gpu" : False,
44 # "epoch" : 20,
45 # "batch_size" : 128,
46 # "maxlen" : 125 , # 超过这个长度的之后字被丢弃，小于这个长度的在前面补空格
47 # "plot_every" : 20, # 每20个batch 可视化一次
48 # "use_env" : True, # 是否使用visodm
49 # "env" : 'poetry' , # visdom env
50 # "max_gen_len" : 200 , # 生成诗歌最长长度
51 # "debug_file" : '/tmp/debugp',
52 # "model_path" : 'checkpoints/tang_199.pth', # 预训练模型路径
53 # "prefix_words": '笑语盈盈暗香去' , # 不是诗歌的组成部分，用来控制生成诗歌的意境
54 # "start_words" : '语' , # 诗歌开始
55 # "acrostic" : False , # 是否是藏头诗
56 # "model_prefix" : 'checkpoints/tang' # 模型保存路径
57 # }
5835 opt = Config()
59 # opt = conf
6036
6137
6238 def generate(model, start_words, ix2word, word2ix, prefix_words=None):
247223 gen_poetry = gen_acrostic if opt.acrostic else generate
248224 result = gen_poetry(model, start_words, ix2word, word2ix, prefix_words)
249225 return result
250 # print(''.join(result))
251226
252227
253228 # if __name__ == '__main__':
254 # conf={
255 # "max_gen_len":200, # 生成诗歌最长长度
256 # "prefix_words":'漂亮', # 不是诗歌的组成部分，用来控制生成诗歌的意境
257 # "start_words" : '雨' # 诗歌开始
229 # conf={'start_words':"雨",'prefix_words':"天晴", 'max_gen_len':100}
230 # start_words = conf['start_words'] #诗歌开始
231 # prefix_words = conf['prefix_words'] #诗歌语境
232 # max_gen_len = conf['max_gen_len'] #诗歌最大长度
233 # cof={
234 # "max_gen_len": max_gen_len, # 生成诗歌最长长度
235 # "prefix_words":prefix_words, # 不是诗歌的组成部分，用来控制生成诗歌的意境
236 # "start_words" : start_words # 诗歌开始
258237 # }
259 # result = gen(**conf)
238 # result = gen(**cof)
260239 # result = ''.join(result)
261 # return {'ret1':result}
262 # # print(result)
240 # # return {'ret1':result}
241 # print(result)