5637e2b
SV08 4 years ago
3 changed file(s) with 5 addition(s) and 12 deletion(s). Raw diff Collapse all Expand all
+2
-9
coding_here.ipynb less more
11 "cells": [
22 {
33 "cell_type": "code",
4 "execution_count": null,
4 "execution_count": 2,
55 "metadata": {},
66 "outputs": [],
77 "source": [
2323 " if __name__ == '__main__':\n",
2424 " result = main.gen(start_words,prefix_words,max_gen_len)\n",
2525 " result = ''.json(result)\n",
26 " return {'你的诗歌': 'result'}\n",
26 " return {'result': result}\n",
2727 " "
2828 ]
29 },
30 {
31 "cell_type": "code",
32 "execution_count": null,
33 "metadata": {},
34 "outputs": [],
35 "source": []
3629 }
3730 ],
3831 "metadata": {
+1
-1
handler.py less more
1515 if __name__ == '__main__':
1616 result = main.gen(start_words,prefix_words,max_gen_len)
1717 result = ''.json(result)
18 return {'你的诗歌': 'result'}
18 return {'result': result}
1919 ⏎
+2
-2
main.py less more
2727 max_gen_len = 200 # 生成诗歌最长长度
2828 debug_file = '/tmp/debugp'
2929 model_path = 'checkpoints/tang_199.pth' # 预训练模型路径
30 prefix_words = '美女' # 不是诗歌的组成部分，用来控制生成诗歌的意境
31 start_words = '雨' # 诗歌开始
30 prefix_words = '笑语盈盈暗香去' # 不是诗歌的组成部分，用来控制生成诗歌的意境
31 start_words = '语' # 诗歌开始
3232 acrostic = False # 是否是藏头诗
3333 model_prefix = 'checkpoints/tang' # 模型保存路径
3434