{"id":60873,"date":"2026-01-16T14:17:33","date_gmt":"2026-01-16T06:17:33","guid":{"rendered":"https:\/\/www.wsisp.com\/helps\/60873.html"},"modified":"2026-01-16T14:17:33","modified_gmt":"2026-01-16T06:17:33","slug":"%e8%87%aa%e7%84%b6%e8%af%ad%e8%a8%80%e5%a4%84%e7%90%86%e5%ae%9e%e6%88%98-%e6%9e%84%e5%bb%ba%e6%99%ba%e8%83%bd%e5%af%b9%e8%af%9d%e7%b3%bb%e7%bb%9f%ef%bc%88bertseq2seq%e6%9e%b6","status":"publish","type":"post","link":"https:\/\/www.wsisp.com\/helps\/60873.html","title":{"rendered":"\u81ea\u7136\u8bed\u8a00\u5904\u7406\u5b9e\u6218\u2014\u2014\u6784\u5efa\u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf\uff08BERT+Seq2Seq\u67b6\u6784\uff09"},"content":{"rendered":"<h2>\u7b2c8\u7bc7&#xff1a;\u81ea\u7136\u8bed\u8a00\u5904\u7406\u5b9e\u6218\u2014\u2014\u6784\u5efa\u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf&#xff08;BERT&#043;Seq2Seq\u67b6\u6784&#xff09;<\/h2>\n<p><img decoding=\"async\" src=\"https:\/\/www.wsisp.com\/helps\/wp-content\/uploads\/2026\/01\/20260116061731-6969d7fbcbf1c.png\" alt=\"\u5728\u8fd9\u91cc\u63d2\u5165\u56fe\u7247\u63cf\u8ff0\" \/><\/p>\n<h3>\u5b66\u4e60\u76ee\u6807<\/h3>\n<li>\u7406\u89e3\u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf\u7684\u6838\u5fc3\u539f\u7406\u4e0e\u67b6\u6784<\/li>\n<li>\u638c\u63e1BERT\u6a21\u578b\u5728\u6587\u672c\u7406\u89e3\u4e2d\u7684\u5e94\u7528\u65b9\u6cd5<\/li>\n<li>\u5b66\u4f1a\u4f7f\u7528Seq2Seq\u6a21\u578b\u5b9e\u73b0\u6587\u672c\u751f\u6210\u529f\u80fd<\/li>\n<li>\u7406\u89e3Attention\u673a\u5236\u5728\u5bf9\u8bdd\u7cfb\u7edf\u4e2d\u7684\u91cd\u8981\u6027<\/li>\n<li>\u80fd\u591f\u72ec\u7acb\u5b8c\u6210\u4e00\u4e2a\u57fa\u4e8eBERT&#043;Seq2Seq\u67b6\u6784\u7684\u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf<\/li>\n<h3>\u7ae0\u8282\u91cd\u70b9<\/h3>\n<li>\u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf\u6982\u8ff0<\/li>\n<li>BERT\u6a21\u578b\u539f\u7406\u4e0e\u5e94\u7528<\/li>\n<li>Seq2Seq\u6a21\u578b\u4e0eAttention\u673a\u5236<\/li>\n<li>\u6570\u636e\u96c6\u51c6\u5907\u4e0e\u9884\u5904\u7406<\/li>\n<li>\u6a21\u578b\u8bad\u7ec3\u4e0e\u4f18\u5316<\/li>\n<li>\u6a21\u578b\u90e8\u7f72\u4e0e\u6d4b\u8bd5<\/li>\n<li>\u6848\u4f8b\u5206\u6790\u4e0e\u4f18\u5316\u601d\u8def<\/li>\n<hr \/>\n<h3>\u4e00\u3001\u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf\u6982\u8ff0<\/h3>\n<h4>1.1 \u4ec0\u4e48\u662f\u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf<\/h4>\n<p>&#x1f4a1; \u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf\u662f\u4e00\u79cd\u80fd\u591f\u901a\u8fc7\u81ea\u7136\u8bed\u8a00\u4e0e\u7528\u6237\u8fdb\u884c\u4ea4\u4e92\u7684\u4eba\u5de5\u667a\u80fd\u7cfb\u7edf&#xff0c;\u5b83\u53ef\u4ee5\u7406\u89e3\u7528\u6237\u7684\u610f\u56fe&#xff0c;\u63d0\u4f9b\u76f8\u5173\u7684\u4fe1\u606f\u6216\u5b8c\u6210\u7279\u5b9a\u7684\u4efb\u52a1\u3002\u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf\u901a\u5e38\u5206\u4e3a\u4e24\u7c7b&#xff1a;<\/p>\n<ul>\n<li>\u95ee\u7b54\u7cfb\u7edf&#xff1a;\u56de\u7b54\u7528\u6237\u7684\u7279\u5b9a\u95ee\u9898&#xff0c;\u5982\u767e\u79d1\u77e5\u8bc6\u95ee\u7b54\u3001\u6280\u672f\u652f\u6301\u95ee\u7b54\u7b49\u3002<\/li>\n<li>\u804a\u5929\u673a\u5668\u4eba&#xff1a;\u8fdb\u884c\u5f00\u653e\u5f0f\u7684\u5bf9\u8bdd&#xff0c;\u5982\u793e\u4ea4\u804a\u5929\u3001\u60c5\u611f\u966a\u4f34\u7b49\u3002<\/li>\n<\/ul>\n<h4>1.2 \u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf\u7684\u6838\u5fc3\u6280\u672f<\/h4>\n<p>\u26a0\ufe0f \u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf\u7684\u6838\u5fc3\u6280\u672f\u5305\u62ec\u81ea\u7136\u8bed\u8a00\u7406\u89e3&#xff08;NLU&#xff09;\u3001\u81ea\u7136\u8bed\u8a00\u751f\u6210&#xff08;NLG&#xff09;\u548c\u5bf9\u8bdd\u7ba1\u7406&#xff08;DM&#xff09;\u3002<\/p>\n<ul>\n<li>\u81ea\u7136\u8bed\u8a00\u7406\u89e3&#xff1a;\u5c06\u7528\u6237\u7684\u81ea\u7136\u8bed\u8a00\u8f93\u5165\u8f6c\u6362\u4e3a\u673a\u5668\u53ef\u7406\u89e3\u7684\u8bed\u4e49\u8868\u793a\u3002<\/li>\n<li>\u81ea\u7136\u8bed\u8a00\u751f\u6210&#xff1a;\u5c06\u673a\u5668\u7684\u8bed\u4e49\u8868\u793a\u8f6c\u6362\u4e3a\u81ea\u7136\u8bed\u8a00\u8f93\u51fa\u3002<\/li>\n<li>\u5bf9\u8bdd\u7ba1\u7406&#xff1a;\u8d1f\u8d23\u5bf9\u8bdd\u7684\u4e0a\u4e0b\u6587\u7406\u89e3\u548c\u72b6\u6001\u7ba1\u7406&#xff0c;\u51b3\u5b9a\u4e0b\u4e00\u6b65\u7684\u56de\u590d\u7b56\u7565\u3002<\/li>\n<\/ul>\n<h4>1.3 \u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf\u7684\u67b6\u6784<\/h4>\n<p>\u2705 \u5e38\u89c1\u7684\u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf\u67b6\u6784\u5206\u4e3a\u4e24\u7c7b&#xff1a;<\/p>\n<ul>\n<li>\u7ba1\u9053\u5f0f\u67b6\u6784&#xff1a;\u5c06NLU\u3001DM\u548cNLG\u5206\u5f00\u5904\u7406&#xff0c;\u6bcf\u4e2a\u7ec4\u4ef6\u72ec\u7acb\u5de5\u4f5c\u3002<\/li>\n<li>\u7aef\u5230\u7aef\u67b6\u6784&#xff1a;\u4f7f\u7528\u6df1\u5ea6\u5b66\u4e60\u6a21\u578b\u76f4\u63a5\u5c06\u7528\u6237\u8f93\u5165\u8f6c\u6362\u4e3a\u7cfb\u7edf\u8f93\u51fa&#xff0c;\u65e0\u9700\u4eba\u5de5\u8bbe\u8ba1\u7684\u4e2d\u95f4\u8868\u793a\u3002<\/li>\n<\/ul>\n<hr \/>\n<h3>\u4e8c\u3001BERT\u6a21\u578b\u539f\u7406\u4e0e\u5e94\u7528<\/h3>\n<h4>2.1 BERT\u6a21\u578b\u6982\u8ff0<\/h4>\n<p>&#x1f4a1; BERT&#xff08;Bidirectional Encoder Representations from Transformers&#xff09;\u662f\u4e00\u79cd\u57fa\u4e8eTransformer\u7684\u9884\u8bad\u7ec3\u8bed\u8a00\u6a21\u578b&#xff0c;\u7531Google\u57282018\u5e74\u63d0\u51fa\u3002BERT\u901a\u8fc7\u53cc\u5411\u4e0a\u4e0b\u6587\u7406\u89e3\u6587\u672c\u7684\u8bed\u4e49\u4fe1\u606f&#xff0c;\u5728\u81ea\u7136\u8bed\u8a00\u5904\u7406\u4efb\u52a1\u4e2d\u53d6\u5f97\u4e86\u663e\u8457\u7684\u6210\u679c\u3002<\/p>\n<h4>2.2 BERT\u7684\u9884\u8bad\u7ec3\u4efb\u52a1<\/h4>\n<p>\u26a0\ufe0f BERT\u7684\u9884\u8bad\u7ec3\u4efb\u52a1\u5305\u62ec\u4e24\u4e2a\u90e8\u5206&#xff1a;<\/p>\n<li>\u63a9\u7801\u8bed\u8a00\u6a21\u578b&#xff08;Masked Language Model&#xff0c;MLM&#xff09;&#xff1a;\u968f\u673a\u906e\u6321\u8f93\u5165\u5e8f\u5217\u4e2d\u7684\u90e8\u5206\u8bcd&#xff0c;\u7136\u540e\u8ba9\u6a21\u578b\u9884\u6d4b\u8fd9\u4e9b\u88ab\u906e\u6321\u7684\u8bcd\u3002<\/li>\n<li>\u4e0b\u4e00\u53e5\u9884\u6d4b&#xff08;Next Sentence Prediction&#xff0c;NSP&#xff09;&#xff1a;\u5224\u65ad\u4e24\u4e2a\u53e5\u5b50\u662f\u5426\u662f\u8fde\u7eed\u7684\u4e0a\u4e0b\u6587\u5173\u7cfb\u3002<\/li>\n<h4>2.3 BERT\u5728\u6587\u672c\u7406\u89e3\u4e2d\u7684\u5e94\u7528<\/h4>\n<p>\u2705 BERT\u5728\u6587\u672c\u7406\u89e3\u4efb\u52a1\u4e2d\u7684\u5e94\u7528\u6b65\u9aa4\u5982\u4e0b&#xff1a;<\/p>\n<li>\u5c06\u6587\u672c\u8f93\u5165\u8f6c\u6362\u4e3aBERT\u53ef\u63a5\u53d7\u7684\u683c\u5f0f\u3002<\/li>\n<li>\u8c03\u7528BERT\u6a21\u578b\u83b7\u53d6\u6587\u672c\u7684\u8bed\u4e49\u8868\u793a\u3002<\/li>\n<li>\u5728BERT\u7684\u8f93\u51fa\u57fa\u7840\u4e0a\u6dfb\u52a0\u7279\u5b9a\u4efb\u52a1\u7684\u5934&#xff08;\u5982\u5206\u7c7b\u5934\u3001\u56de\u5f52\u5934\u7b49&#xff09;\u3002<\/li>\n<li>\u5bf9\u6a21\u578b\u8fdb\u884c\u5fae\u8c03&#xff0c;\u4ee5\u9002\u5e94\u7279\u5b9a\u4efb\u52a1\u3002<\/li>\n<h4>2.4 \u4ee3\u7801\u5b9e\u73b0&#xff1a;\u4f7f\u7528BERT\u8fdb\u884c\u6587\u672c\u5206\u7c7b<\/h4>\n<p><span class=\"token keyword\">import<\/span> torch<br \/>\n<span class=\"token keyword\">from<\/span> transformers <span class=\"token keyword\">import<\/span> BertTokenizer<span class=\"token punctuation\">,<\/span> BertForSequenceClassification<\/p>\n<p><span class=\"token comment\"># \u52a0\u8f7dBERT\u5206\u8bcd\u5668\u548c\u6a21\u578b<\/span><br \/>\ntokenizer <span class=\"token operator\">&#061;<\/span> BertTokenizer<span class=\"token punctuation\">.<\/span>from_pretrained<span class=\"token punctuation\">(<\/span><span class=\"token string\">&#039;bert-base-chinese&#039;<\/span><span class=\"token punctuation\">)<\/span><br \/>\nmodel <span class=\"token operator\">&#061;<\/span> BertForSequenceClassification<span class=\"token punctuation\">.<\/span>from_pretrained<span class=\"token punctuation\">(<\/span><span class=\"token string\">&#039;bert-base-chinese&#039;<\/span><span class=\"token punctuation\">,<\/span> num_labels<span class=\"token operator\">&#061;<\/span><span class=\"token number\">2<\/span><span class=\"token punctuation\">)<\/span><\/p>\n<p><span class=\"token comment\"># \u6587\u672c\u8f93\u5165<\/span><br \/>\ntext <span class=\"token operator\">&#061;<\/span> <span class=\"token string\">&#034;\u8fd9\u662f\u4e00\u4e2a\u6d4b\u8bd5\u6587\u672c&#034;<\/span><\/p>\n<p><span class=\"token comment\"># \u6587\u672c\u9884\u5904\u7406<\/span><br \/>\ninputs <span class=\"token operator\">&#061;<\/span> tokenizer<span class=\"token punctuation\">(<\/span>text<span class=\"token punctuation\">,<\/span> return_tensors<span class=\"token operator\">&#061;<\/span><span class=\"token string\">&#039;pt&#039;<\/span><span class=\"token punctuation\">,<\/span> padding<span class=\"token operator\">&#061;<\/span><span class=\"token boolean\">True<\/span><span class=\"token punctuation\">,<\/span> truncation<span class=\"token operator\">&#061;<\/span><span class=\"token boolean\">True<\/span><span class=\"token punctuation\">,<\/span> max_length<span class=\"token operator\">&#061;<\/span><span class=\"token number\">512<\/span><span class=\"token punctuation\">)<\/span><\/p>\n<p><span class=\"token comment\"># \u6a21\u578b\u63a8\u7406<\/span><br \/>\n<span class=\"token keyword\">with<\/span> torch<span class=\"token punctuation\">.<\/span>no_grad<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span><br \/>\n    outputs <span class=\"token operator\">&#061;<\/span> model<span class=\"token punctuation\">(<\/span><span class=\"token operator\">**<\/span>inputs<span class=\"token punctuation\">)<\/span><\/p>\n<p><span class=\"token comment\"># \u83b7\u53d6\u9884\u6d4b\u7ed3\u679c<\/span><br \/>\nlogits <span class=\"token operator\">&#061;<\/span> outputs<span class=\"token punctuation\">.<\/span>logits<br \/>\npredicted_label <span class=\"token operator\">&#061;<\/span> torch<span class=\"token punctuation\">.<\/span>argmax<span class=\"token punctuation\">(<\/span>logits<span class=\"token punctuation\">,<\/span> dim<span class=\"token operator\">&#061;<\/span><span class=\"token number\">1<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">.<\/span>item<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><\/p>\n<p><span class=\"token keyword\">print<\/span><span class=\"token punctuation\">(<\/span><span class=\"token string-interpolation\"><span class=\"token string\">f&#034;\u9884\u6d4b\u6807\u7b7e&#xff1a;<\/span><span class=\"token interpolation\"><span class=\"token punctuation\">{<\/span>predicted_label<span class=\"token punctuation\">}<\/span><\/span><span class=\"token string\">&#034;<\/span><\/span><span class=\"token punctuation\">)<\/span><\/p>\n<hr \/>\n<h3>\u4e09\u3001Seq2Seq\u6a21\u578b\u4e0eAttention\u673a\u5236<\/h3>\n<h4>3.1 Seq2Seq\u6a21\u578b\u6982\u8ff0<\/h4>\n<p>&#x1f4a1; Seq2Seq&#xff08;Sequence to Sequence&#xff09;\u6a21\u578b\u662f\u4e00\u79cd\u7528\u4e8e\u5904\u7406\u5e8f\u5217\u6570\u636e\u7684\u6df1\u5ea6\u5b66\u4e60\u6a21\u578b&#xff0c;\u7531\u7f16\u7801\u5668&#xff08;Encoder&#xff09;\u548c\u89e3\u7801\u5668&#xff08;Decoder&#xff09;\u7ec4\u6210\u3002\u7f16\u7801\u5668\u5c06\u8f93\u5165\u5e8f\u5217\u8f6c\u6362\u4e3a\u56fa\u5b9a\u957f\u5ea6\u7684\u5411\u91cf\u8868\u793a&#xff0c;\u89e3\u7801\u5668\u6839\u636e\u8be5\u5411\u91cf\u8868\u793a\u751f\u6210\u8f93\u51fa\u5e8f\u5217\u3002<\/p>\n<h4>3.2 Seq2Seq\u6a21\u578b\u7684\u5c40\u9650\u6027<\/h4>\n<p>\u26a0\ufe0f \u4f20\u7edf\u7684Seq2Seq\u6a21\u578b\u5b58\u5728\u4e00\u4e2a\u660e\u663e\u7684\u5c40\u9650\u6027&#xff1a;\u5f53\u8f93\u5165\u5e8f\u5217\u8f83\u957f\u65f6&#xff0c;\u7f16\u7801\u5668\u65e0\u6cd5\u5c06\u6240\u6709\u4fe1\u606f\u538b\u7f29\u5230\u4e00\u4e2a\u56fa\u5b9a\u957f\u5ea6\u7684\u5411\u91cf\u4e2d&#xff0c;\u5bfc\u81f4\u89e3\u7801\u5668\u65e0\u6cd5\u751f\u6210\u9ad8\u8d28\u91cf\u7684\u8f93\u51fa\u3002<\/p>\n<h4>3.3 Attention\u673a\u5236<\/h4>\n<p>\u2705 Attention\u673a\u5236\u662f\u4e00\u79cd\u89e3\u51b3Seq2Seq\u6a21\u578b\u5c40\u9650\u6027\u7684\u65b9\u6cd5&#xff0c;\u5b83\u5141\u8bb8\u89e3\u7801\u5668\u5728\u751f\u6210\u6bcf\u4e2a\u8f93\u51fa\u8bcd\u65f6&#xff0c;\u5173\u6ce8\u8f93\u5165\u5e8f\u5217\u4e2d\u4e0e\u8be5\u8bcd\u76f8\u5173\u7684\u90e8\u5206\u3002Attention\u673a\u5236\u7684\u6838\u5fc3\u601d\u60f3\u662f\u8ba1\u7b97\u8f93\u5165\u5e8f\u5217\u4e2d\u6bcf\u4e2a\u4f4d\u7f6e\u7684\u6743\u91cd&#xff0c;\u7136\u540e\u6839\u636e\u6743\u91cd\u5bf9\u8f93\u5165\u5e8f\u5217\u7684\u8868\u793a\u8fdb\u884c\u52a0\u6743\u6c42\u548c\u3002<\/p>\n<h4>3.4 \u4ee3\u7801\u5b9e\u73b0&#xff1a;\u7b80\u5355\u7684Seq2Seq\u6a21\u578b<\/h4>\n<p><span class=\"token keyword\">import<\/span> torch<br \/>\n<span class=\"token keyword\">import<\/span> torch<span class=\"token punctuation\">.<\/span>nn <span class=\"token keyword\">as<\/span> nn<br \/>\n<span class=\"token keyword\">import<\/span> torch<span class=\"token punctuation\">.<\/span>optim <span class=\"token keyword\">as<\/span> optim<\/p>\n<p><span class=\"token comment\"># \u5b9a\u4e49Seq2Seq\u6a21\u578b<\/span><br \/>\n<span class=\"token keyword\">class<\/span> <span class=\"token class-name\">Seq2Seq<\/span><span class=\"token punctuation\">(<\/span>nn<span class=\"token punctuation\">.<\/span>Module<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span><br \/>\n    <span class=\"token keyword\">def<\/span> <span class=\"token function\">__init__<\/span><span class=\"token punctuation\">(<\/span>self<span class=\"token punctuation\">,<\/span> input_size<span class=\"token punctuation\">,<\/span> hidden_size<span class=\"token punctuation\">,<\/span> output_size<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span><br \/>\n        <span class=\"token builtin\">super<\/span><span class=\"token punctuation\">(<\/span>Seq2Seq<span class=\"token punctuation\">,<\/span> self<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">.<\/span>__init__<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><br \/>\n        self<span class=\"token punctuation\">.<\/span>encoder <span class=\"token operator\">&#061;<\/span> nn<span class=\"token punctuation\">.<\/span>LSTM<span class=\"token punctuation\">(<\/span>input_size<span class=\"token punctuation\">,<\/span> hidden_size<span class=\"token punctuation\">,<\/span> batch_first<span class=\"token operator\">&#061;<\/span><span class=\"token boolean\">True<\/span><span class=\"token punctuation\">)<\/span><br \/>\n        self<span class=\"token punctuation\">.<\/span>decoder <span class=\"token operator\">&#061;<\/span> nn<span class=\"token punctuation\">.<\/span>LSTM<span class=\"token punctuation\">(<\/span>hidden_size<span class=\"token punctuation\">,<\/span> hidden_size<span class=\"token punctuation\">,<\/span> batch_first<span class=\"token operator\">&#061;<\/span><span class=\"token boolean\">True<\/span><span class=\"token punctuation\">)<\/span><br \/>\n        self<span class=\"token punctuation\">.<\/span>fc <span class=\"token operator\">&#061;<\/span> nn<span class=\"token punctuation\">.<\/span>Linear<span class=\"token punctuation\">(<\/span>hidden_size<span class=\"token punctuation\">,<\/span> output_size<span class=\"token punctuation\">)<\/span><\/p>\n<p>    <span class=\"token keyword\">def<\/span> <span class=\"token function\">forward<\/span><span class=\"token punctuation\">(<\/span>self<span class=\"token punctuation\">,<\/span> input_seq<span class=\"token punctuation\">,<\/span> target_seq<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span><br \/>\n        <span class=\"token comment\"># \u7f16\u7801\u5668<\/span><br \/>\n        encoder_output<span class=\"token punctuation\">,<\/span> <span class=\"token punctuation\">(<\/span>encoder_hidden<span class=\"token punctuation\">,<\/span> encoder_cell<span class=\"token punctuation\">)<\/span> <span class=\"token operator\">&#061;<\/span> self<span class=\"token punctuation\">.<\/span>encoder<span class=\"token punctuation\">(<\/span>input_seq<span class=\"token punctuation\">)<\/span><\/p>\n<p>        <span class=\"token comment\"># \u89e3\u7801\u5668<\/span><br \/>\n        decoder_output<span class=\"token punctuation\">,<\/span> <span class=\"token punctuation\">(<\/span>decoder_hidden<span class=\"token punctuation\">,<\/span> decoder_cell<span class=\"token punctuation\">)<\/span> <span class=\"token operator\">&#061;<\/span> self<span class=\"token punctuation\">.<\/span>decoder<span class=\"token punctuation\">(<\/span>target_seq<span class=\"token punctuation\">,<\/span> <span class=\"token punctuation\">(<\/span>encoder_hidden<span class=\"token punctuation\">,<\/span> encoder_cell<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">)<\/span><\/p>\n<p>        <span class=\"token comment\"># \u8f93\u51fa\u5c42<\/span><br \/>\n        output <span class=\"token operator\">&#061;<\/span> self<span class=\"token punctuation\">.<\/span>fc<span class=\"token punctuation\">(<\/span>decoder_output<span class=\"token punctuation\">)<\/span><\/p>\n<p>        <span class=\"token keyword\">return<\/span> output<\/p>\n<p><span class=\"token comment\"># \u8d85\u53c2\u6570\u8bbe\u7f6e<\/span><br \/>\ninput_size <span class=\"token operator\">&#061;<\/span> <span class=\"token number\">10<\/span><br \/>\nhidden_size <span class=\"token operator\">&#061;<\/span> <span class=\"token number\">20<\/span><br \/>\noutput_size <span class=\"token operator\">&#061;<\/span> <span class=\"token number\">10<\/span><br \/>\nbatch_size <span class=\"token operator\">&#061;<\/span> <span class=\"token number\">2<\/span><br \/>\nseq_length <span class=\"token operator\">&#061;<\/span> <span class=\"token number\">5<\/span><\/p>\n<p><span class=\"token comment\"># \u751f\u6210\u6a21\u62df\u6570\u636e<\/span><br \/>\ninput_seq <span class=\"token operator\">&#061;<\/span> torch<span class=\"token punctuation\">.<\/span>randn<span class=\"token punctuation\">(<\/span>batch_size<span class=\"token punctuation\">,<\/span> seq_length<span class=\"token punctuation\">,<\/span> input_size<span class=\"token punctuation\">)<\/span><br \/>\ntarget_seq <span class=\"token operator\">&#061;<\/span> torch<span class=\"token punctuation\">.<\/span>randn<span class=\"token punctuation\">(<\/span>batch_size<span class=\"token punctuation\">,<\/span> seq_length<span class=\"token punctuation\">,<\/span> output_size<span class=\"token punctuation\">)<\/span><\/p>\n<p><span class=\"token comment\"># \u6a21\u578b\u5b9e\u4f8b\u5316<\/span><br \/>\nmodel <span class=\"token operator\">&#061;<\/span> Seq2Seq<span class=\"token punctuation\">(<\/span>input_size<span class=\"token punctuation\">,<\/span> hidden_size<span class=\"token punctuation\">,<\/span> output_size<span class=\"token punctuation\">)<\/span><\/p>\n<p><span class=\"token comment\"># \u635f\u5931\u51fd\u6570\u548c\u4f18\u5316\u5668<\/span><br \/>\ncriterion <span class=\"token operator\">&#061;<\/span> nn<span class=\"token punctuation\">.<\/span>MSELoss<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><br \/>\noptimizer <span class=\"token operator\">&#061;<\/span> optim<span class=\"token punctuation\">.<\/span>Adam<span class=\"token punctuation\">(<\/span>model<span class=\"token punctuation\">.<\/span>parameters<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">,<\/span> lr<span class=\"token operator\">&#061;<\/span><span class=\"token number\">0.001<\/span><span class=\"token punctuation\">)<\/span><\/p>\n<p><span class=\"token comment\"># \u8bad\u7ec3\u8fc7\u7a0b<\/span><br \/>\nmodel<span class=\"token punctuation\">.<\/span>train<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><br \/>\n<span class=\"token keyword\">for<\/span> epoch <span class=\"token keyword\">in<\/span> <span class=\"token builtin\">range<\/span><span class=\"token punctuation\">(<\/span><span class=\"token number\">100<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span><br \/>\n    optimizer<span class=\"token punctuation\">.<\/span>zero_grad<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><br \/>\n    output <span class=\"token operator\">&#061;<\/span> model<span class=\"token punctuation\">(<\/span>input_seq<span class=\"token punctuation\">,<\/span> target_seq<span class=\"token punctuation\">)<\/span><br \/>\n    loss <span class=\"token operator\">&#061;<\/span> criterion<span class=\"token punctuation\">(<\/span>output<span class=\"token punctuation\">,<\/span> target_seq<span class=\"token punctuation\">)<\/span><br \/>\n    loss<span class=\"token punctuation\">.<\/span>backward<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><br \/>\n    optimizer<span class=\"token punctuation\">.<\/span>step<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><\/p>\n<p>    <span class=\"token keyword\">if<\/span> <span class=\"token punctuation\">(<\/span>epoch <span class=\"token operator\">&#043;<\/span> <span class=\"token number\">1<\/span><span class=\"token punctuation\">)<\/span> <span class=\"token operator\">%<\/span> <span class=\"token number\">10<\/span> <span class=\"token operator\">&#061;&#061;<\/span> <span class=\"token number\">0<\/span><span class=\"token punctuation\">:<\/span><br \/>\n        <span class=\"token keyword\">print<\/span><span class=\"token punctuation\">(<\/span><span class=\"token string-interpolation\"><span class=\"token string\">f&#034;Epoch: <\/span><span class=\"token interpolation\"><span class=\"token punctuation\">{<\/span>epoch <span class=\"token operator\">&#043;<\/span> <span class=\"token number\">1<\/span><span class=\"token punctuation\">}<\/span><\/span><span class=\"token string\">, Loss: <\/span><span class=\"token interpolation\"><span class=\"token punctuation\">{<\/span>loss<span class=\"token punctuation\">.<\/span>item<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span><span class=\"token format-spec\">.4f<\/span><span class=\"token punctuation\">}<\/span><\/span><span class=\"token string\">&#034;<\/span><\/span><span class=\"token punctuation\">)<\/span><\/p>\n<p><span class=\"token comment\"># \u6d4b\u8bd5\u8fc7\u7a0b<\/span><br \/>\nmodel<span class=\"token punctuation\">.<\/span><span class=\"token builtin\">eval<\/span><span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><br \/>\n<span class=\"token keyword\">with<\/span> torch<span class=\"token punctuation\">.<\/span>no_grad<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span><br \/>\n    output <span class=\"token operator\">&#061;<\/span> model<span class=\"token punctuation\">(<\/span>input_seq<span class=\"token punctuation\">,<\/span> target_seq<span class=\"token punctuation\">)<\/span><br \/>\n    <span class=\"token keyword\">print<\/span><span class=\"token punctuation\">(<\/span><span class=\"token string-interpolation\"><span class=\"token string\">f&#034;\u6d4b\u8bd5\u8f93\u51fa&#xff1a;<\/span><span class=\"token interpolation\"><span class=\"token punctuation\">{<\/span>output<span class=\"token punctuation\">}<\/span><\/span><span class=\"token string\">&#034;<\/span><\/span><span class=\"token punctuation\">)<\/span><\/p>\n<hr \/>\n<h3>\u56db\u3001\u6570\u636e\u96c6\u51c6\u5907\u4e0e\u9884\u5904\u7406<\/h3>\n<h4>4.1 \u6570\u636e\u96c6\u9009\u62e9<\/h4>\n<p>&#x1f4a1; \u9009\u62e9\u5408\u9002\u7684\u6570\u636e\u96c6\u662f\u6784\u5efa\u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf\u7684\u91cd\u8981\u6b65\u9aa4\u3002\u5e38\u89c1\u7684\u5bf9\u8bdd\u6570\u636e\u96c6\u5305\u62ec&#xff1a;<\/p>\n<ul>\n<li>Cornell Movie-Dialogs Corpus&#xff1a;\u5305\u542b\u7535\u5f71\u4e2d\u7684\u5bf9\u8bdd\u6570\u636e\u3002<\/li>\n<li>DailyDialog&#xff1a;\u5305\u542b\u65e5\u5e38\u751f\u6d3b\u4e2d\u7684\u5bf9\u8bdd\u6570\u636e\u3002<\/li>\n<li>Chinese Dialog Corpus&#xff1a;\u5305\u542b\u4e2d\u6587\u5bf9\u8bdd\u6570\u636e\u3002<\/li>\n<\/ul>\n<h4>4.2 \u6570\u636e\u9884\u5904\u7406\u6b65\u9aa4<\/h4>\n<p>\u26a0\ufe0f \u6570\u636e\u9884\u5904\u7406\u662f\u6784\u5efa\u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf\u7684\u5173\u952e\u6b65\u9aa4&#xff0c;\u4e3b\u8981\u5305\u62ec\u4ee5\u4e0b\u5185\u5bb9&#xff1a;<\/p>\n<li>\u6570\u636e\u6e05\u6d17&#xff1a;\u53bb\u9664\u566a\u58f0\u6570\u636e&#xff0c;\u5982\u8868\u60c5\u7b26\u53f7\u3001\u7279\u6b8a\u5b57\u7b26\u7b49\u3002<\/li>\n<li>\u5206\u8bcd&#xff1a;\u5c06\u6587\u672c\u5206\u5272\u4e3a\u5355\u8bcd\u6216\u5b50\u8bcd\u3002<\/li>\n<li>\u6784\u5efa\u8bcd\u6c47\u8868&#xff1a;\u7edf\u8ba1\u6587\u672c\u4e2d\u7684\u5355\u8bcd\u9891\u7387&#xff0c;\u6784\u5efa\u8bcd\u6c47\u8868\u3002<\/li>\n<li>\u5e8f\u5217\u586b\u5145&#xff1a;\u5c06\u8f93\u5165\u548c\u8f93\u51fa\u5e8f\u5217\u586b\u5145\u5230\u76f8\u540c\u7684\u957f\u5ea6&#xff0c;\u4ee5\u4fbf\u4e8e\u6a21\u578b\u8bad\u7ec3\u3002<\/li>\n<h4>4.3 \u4ee3\u7801\u5b9e\u73b0&#xff1a;\u6570\u636e\u9884\u5904\u7406<\/h4>\n<p><span class=\"token keyword\">import<\/span> torch<br \/>\n<span class=\"token keyword\">from<\/span> transformers <span class=\"token keyword\">import<\/span> BertTokenizer<br \/>\n<span class=\"token keyword\">import<\/span> pandas <span class=\"token keyword\">as<\/span> pd<\/p>\n<p><span class=\"token comment\"># \u52a0\u8f7d\u6570\u636e\u96c6<\/span><br \/>\ndf <span class=\"token operator\">&#061;<\/span> pd<span class=\"token punctuation\">.<\/span>read_csv<span class=\"token punctuation\">(<\/span><span class=\"token string\">&#039;dialog_data.csv&#039;<\/span><span class=\"token punctuation\">,<\/span> names<span class=\"token operator\">&#061;<\/span><span class=\"token punctuation\">[<\/span><span class=\"token string\">&#039;context&#039;<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token string\">&#039;response&#039;<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">)<\/span><\/p>\n<p><span class=\"token comment\"># \u52a0\u8f7dBERT\u5206\u8bcd\u5668<\/span><br \/>\ntokenizer <span class=\"token operator\">&#061;<\/span> BertTokenizer<span class=\"token punctuation\">.<\/span>from_pretrained<span class=\"token punctuation\">(<\/span><span class=\"token string\">&#039;bert-base-chinese&#039;<\/span><span class=\"token punctuation\">)<\/span><\/p>\n<p><span class=\"token comment\"># \u6570\u636e\u9884\u5904\u7406\u51fd\u6570<\/span><br \/>\n<span class=\"token keyword\">def<\/span> <span class=\"token function\">preprocess_data<\/span><span class=\"token punctuation\">(<\/span>context<span class=\"token punctuation\">,<\/span> response<span class=\"token punctuation\">,<\/span> tokenizer<span class=\"token punctuation\">,<\/span> max_length<span class=\"token operator\">&#061;<\/span><span class=\"token number\">512<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span><br \/>\n    <span class=\"token comment\"># \u5bf9\u4e0a\u4e0b\u6587\u8fdb\u884c\u7f16\u7801<\/span><br \/>\n    context_encoding <span class=\"token operator\">&#061;<\/span> tokenizer<span class=\"token punctuation\">(<\/span><br \/>\n        context<span class=\"token punctuation\">,<\/span><br \/>\n        padding<span class=\"token operator\">&#061;<\/span><span class=\"token string\">&#039;max_length&#039;<\/span><span class=\"token punctuation\">,<\/span><br \/>\n        truncation<span class=\"token operator\">&#061;<\/span><span class=\"token boolean\">True<\/span><span class=\"token punctuation\">,<\/span><br \/>\n        max_length<span class=\"token operator\">&#061;<\/span>max_length<span class=\"token punctuation\">,<\/span><br \/>\n        return_tensors<span class=\"token operator\">&#061;<\/span><span class=\"token string\">&#039;pt&#039;<\/span><br \/>\n    <span class=\"token punctuation\">)<\/span><\/p>\n<p>    <span class=\"token comment\"># \u5bf9\u56de\u590d\u8fdb\u884c\u7f16\u7801<\/span><br \/>\n    response_encoding <span class=\"token operator\">&#061;<\/span> tokenizer<span class=\"token punctuation\">(<\/span><br \/>\n        response<span class=\"token punctuation\">,<\/span><br \/>\n        padding<span class=\"token operator\">&#061;<\/span><span class=\"token string\">&#039;max_length&#039;<\/span><span class=\"token punctuation\">,<\/span><br \/>\n        truncation<span class=\"token operator\">&#061;<\/span><span class=\"token boolean\">True<\/span><span class=\"token punctuation\">,<\/span><br \/>\n        max_length<span class=\"token operator\">&#061;<\/span>max_length<span class=\"token punctuation\">,<\/span><br \/>\n        return_tensors<span class=\"token operator\">&#061;<\/span><span class=\"token string\">&#039;pt&#039;<\/span><br \/>\n    <span class=\"token punctuation\">)<\/span><\/p>\n<p>    <span class=\"token keyword\">return<\/span> <span class=\"token punctuation\">{<\/span><br \/>\n        <span class=\"token string\">&#039;context_input_ids&#039;<\/span><span class=\"token punctuation\">:<\/span> context_encoding<span class=\"token punctuation\">[<\/span><span class=\"token string\">&#039;input_ids&#039;<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">,<\/span><br \/>\n        <span class=\"token string\">&#039;context_attention_mask&#039;<\/span><span class=\"token punctuation\">:<\/span> context_encoding<span class=\"token punctuation\">[<\/span><span class=\"token string\">&#039;attention_mask&#039;<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">,<\/span><br \/>\n        <span class=\"token string\">&#039;response_input_ids&#039;<\/span><span class=\"token punctuation\">:<\/span> response_encoding<span class=\"token punctuation\">[<\/span><span class=\"token string\">&#039;input_ids&#039;<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">,<\/span><br \/>\n        <span class=\"token string\">&#039;response_attention_mask&#039;<\/span><span class=\"token punctuation\">:<\/span> response_encoding<span class=\"token punctuation\">[<\/span><span class=\"token string\">&#039;attention_mask&#039;<\/span><span class=\"token punctuation\">]<\/span><br \/>\n    <span class=\"token punctuation\">}<\/span><\/p>\n<p><span class=\"token comment\"># \u5e94\u7528\u6570\u636e\u9884\u5904\u7406\u51fd\u6570<\/span><br \/>\nprocessed_data <span class=\"token operator\">&#061;<\/span> <span class=\"token punctuation\">[<\/span><span class=\"token punctuation\">]<\/span><br \/>\n<span class=\"token keyword\">for<\/span> index<span class=\"token punctuation\">,<\/span> row <span class=\"token keyword\">in<\/span> df<span class=\"token punctuation\">.<\/span>iterrows<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span><br \/>\n    processed_data<span class=\"token punctuation\">.<\/span>append<span class=\"token punctuation\">(<\/span><br \/>\n        preprocess_data<span class=\"token punctuation\">(<\/span>row<span class=\"token punctuation\">[<\/span><span class=\"token string\">&#039;context&#039;<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">,<\/span> row<span class=\"token punctuation\">[<\/span><span class=\"token string\">&#039;response&#039;<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">,<\/span> tokenizer<span class=\"token punctuation\">)<\/span><br \/>\n    <span class=\"token punctuation\">)<\/span><\/p>\n<p><span class=\"token comment\"># \u5c06\u5904\u7406\u540e\u7684\u6570\u636e\u8f6c\u6362\u4e3a\u5f20\u91cf<\/span><br \/>\ncontext_input_ids <span class=\"token operator\">&#061;<\/span> torch<span class=\"token punctuation\">.<\/span>cat<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">[<\/span>data<span class=\"token punctuation\">[<\/span><span class=\"token string\">&#039;context_input_ids&#039;<\/span><span class=\"token punctuation\">]<\/span> <span class=\"token keyword\">for<\/span> data <span class=\"token keyword\">in<\/span> processed_data<span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">)<\/span><br \/>\ncontext_attention_mask <span class=\"token operator\">&#061;<\/span> torch<span class=\"token punctuation\">.<\/span>cat<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">[<\/span>data<span class=\"token punctuation\">[<\/span><span class=\"token string\">&#039;context_attention_mask&#039;<\/span><span class=\"token punctuation\">]<\/span> <span class=\"token keyword\">for<\/span> data <span class=\"token keyword\">in<\/span> processed_data<span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">)<\/span><br \/>\nresponse_input_ids <span class=\"token operator\">&#061;<\/span> torch<span class=\"token punctuation\">.<\/span>cat<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">[<\/span>data<span class=\"token punctuation\">[<\/span><span class=\"token string\">&#039;response_input_ids&#039;<\/span><span class=\"token punctuation\">]<\/span> <span class=\"token keyword\">for<\/span> data <span class=\"token keyword\">in<\/span> processed_data<span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">)<\/span><br \/>\nresponse_attention_mask <span class=\"token operator\">&#061;<\/span> torch<span class=\"token punctuation\">.<\/span>cat<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">[<\/span>data<span class=\"token punctuation\">[<\/span><span class=\"token string\">&#039;response_attention_mask&#039;<\/span><span class=\"token punctuation\">]<\/span> <span class=\"token keyword\">for<\/span> data <span class=\"token keyword\">in<\/span> processed_data<span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">)<\/span><\/p>\n<p><span class=\"token comment\"># \u4fdd\u5b58\u5904\u7406\u540e\u7684\u6570\u636e<\/span><br \/>\ntorch<span class=\"token punctuation\">.<\/span>save<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">{<\/span><br \/>\n    <span class=\"token string\">&#039;context_input_ids&#039;<\/span><span class=\"token punctuation\">:<\/span> context_input_ids<span class=\"token punctuation\">,<\/span><br \/>\n    <span class=\"token string\">&#039;context_attention_mask&#039;<\/span><span class=\"token punctuation\">:<\/span> context_attention_mask<span class=\"token punctuation\">,<\/span><br \/>\n    <span class=\"token string\">&#039;response_input_ids&#039;<\/span><span class=\"token punctuation\">:<\/span> response_input_ids<span class=\"token punctuation\">,<\/span><br \/>\n    <span class=\"token string\">&#039;response_attention_mask&#039;<\/span><span class=\"token punctuation\">:<\/span> response_attention_mask<br \/>\n<span class=\"token punctuation\">}<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token string\">&#039;processed_data.pt&#039;<\/span><span class=\"token punctuation\">)<\/span><\/p>\n<hr \/>\n<h3>\u4e94\u3001\u6a21\u578b\u8bad\u7ec3\u4e0e\u4f18\u5316<\/h3>\n<h4>5.1 \u6a21\u578b\u67b6\u6784\u8bbe\u8ba1<\/h4>\n<p>&#x1f4a1; \u6211\u4eec\u5c06\u4f7f\u7528BERT\u4f5c\u4e3a\u7f16\u7801\u5668&#xff0c;Seq2Seq\u6a21\u578b\u4f5c\u4e3a\u89e3\u7801\u5668&#xff0c;\u6784\u5efa\u4e00\u4e2aBERT&#043;Seq2Seq\u67b6\u6784\u7684\u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf\u3002BERT\u8d1f\u8d23\u7406\u89e3\u8f93\u5165\u5e8f\u5217\u7684\u8bed\u4e49\u4fe1\u606f&#xff0c;Seq2Seq\u6a21\u578b\u8d1f\u8d23\u751f\u6210\u8f93\u51fa\u5e8f\u5217\u3002<\/p>\n<h4>5.2 \u635f\u5931\u51fd\u6570\u4e0e\u4f18\u5316\u5668<\/h4>\n<p>\u26a0\ufe0f \u5bf9\u4e8e\u6587\u672c\u751f\u6210\u4efb\u52a1&#xff0c;\u5e38\u7528\u7684\u635f\u5931\u51fd\u6570\u662f\u4ea4\u53c9\u71b5\u635f\u5931\u51fd\u6570\u3002\u4f18\u5316\u5668\u53ef\u4ee5\u9009\u62e9Adam\u6216SGD\u7b49\u3002<\/p>\n<h4>5.3 \u8bad\u7ec3\u8fc7\u7a0b<\/h4>\n<p>\u2705 \u8bad\u7ec3\u8fc7\u7a0b\u4e3b\u8981\u5305\u62ec\u4ee5\u4e0b\u6b65\u9aa4&#xff1a;<\/p>\n<li>\u52a0\u8f7d\u9884\u5904\u7406\u540e\u7684\u6570\u636e\u96c6\u3002<\/li>\n<li>\u521d\u59cb\u5316\u6a21\u578b\u3001\u635f\u5931\u51fd\u6570\u548c\u4f18\u5316\u5668\u3002<\/li>\n<li>\u8fed\u4ee3\u8bad\u7ec3\u6a21\u578b&#xff0c;\u8ba1\u7b97\u635f\u5931\u5e76\u66f4\u65b0\u53c2\u6570\u3002<\/li>\n<li>\u5b9a\u671f\u4fdd\u5b58\u6a21\u578b\u68c0\u67e5\u70b9&#xff0c;\u4ee5\u4fbf\u4e8e\u540e\u7eed\u8bc4\u4f30\u548c\u90e8\u7f72\u3002<\/li>\n<h4>5.4 \u4ee3\u7801\u5b9e\u73b0&#xff1a;\u6a21\u578b\u8bad\u7ec3<\/h4>\n<p><span class=\"token keyword\">import<\/span> torch<br \/>\n<span class=\"token keyword\">import<\/span> torch<span class=\"token punctuation\">.<\/span>nn <span class=\"token keyword\">as<\/span> nn<br \/>\n<span class=\"token keyword\">import<\/span> torch<span class=\"token punctuation\">.<\/span>optim <span class=\"token keyword\">as<\/span> optim<br \/>\n<span class=\"token keyword\">from<\/span> transformers <span class=\"token keyword\">import<\/span> BertModel<span class=\"token punctuation\">,<\/span> BertTokenizer<\/p>\n<p><span class=\"token comment\"># \u5b9a\u4e49BERT&#043;Seq2Seq\u6a21\u578b<\/span><br \/>\n<span class=\"token keyword\">class<\/span> <span class=\"token class-name\">BERTSeq2Seq<\/span><span class=\"token punctuation\">(<\/span>nn<span class=\"token punctuation\">.<\/span>Module<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span><br \/>\n    <span class=\"token keyword\">def<\/span> <span class=\"token function\">__init__<\/span><span class=\"token punctuation\">(<\/span>self<span class=\"token punctuation\">,<\/span> bert_model_name<span class=\"token punctuation\">,<\/span> hidden_size<span class=\"token punctuation\">,<\/span> output_size<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span><br \/>\n        <span class=\"token builtin\">super<\/span><span class=\"token punctuation\">(<\/span>BERTSeq2Seq<span class=\"token punctuation\">,<\/span> self<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">.<\/span>__init__<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><br \/>\n        self<span class=\"token punctuation\">.<\/span>bert <span class=\"token operator\">&#061;<\/span> BertModel<span class=\"token punctuation\">.<\/span>from_pretrained<span class=\"token punctuation\">(<\/span>bert_model_name<span class=\"token punctuation\">)<\/span><br \/>\n        self<span class=\"token punctuation\">.<\/span>decoder <span class=\"token operator\">&#061;<\/span> nn<span class=\"token punctuation\">.<\/span>LSTM<span class=\"token punctuation\">(<\/span>hidden_size<span class=\"token punctuation\">,<\/span> hidden_size<span class=\"token punctuation\">,<\/span> batch_first<span class=\"token operator\">&#061;<\/span><span class=\"token boolean\">True<\/span><span class=\"token punctuation\">)<\/span><br \/>\n        self<span class=\"token punctuation\">.<\/span>fc <span class=\"token operator\">&#061;<\/span> nn<span class=\"token punctuation\">.<\/span>Linear<span class=\"token punctuation\">(<\/span>hidden_size<span class=\"token punctuation\">,<\/span> output_size<span class=\"token punctuation\">)<\/span><\/p>\n<p>    <span class=\"token keyword\">def<\/span> <span class=\"token function\">forward<\/span><span class=\"token punctuation\">(<\/span>self<span class=\"token punctuation\">,<\/span> context_input_ids<span class=\"token punctuation\">,<\/span> context_attention_mask<span class=\"token punctuation\">,<\/span> response_input_ids<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span><br \/>\n        <span class=\"token comment\"># BERT\u7f16\u7801<\/span><br \/>\n        bert_output <span class=\"token operator\">&#061;<\/span> self<span class=\"token punctuation\">.<\/span>bert<span class=\"token punctuation\">(<\/span><br \/>\n            input_ids<span class=\"token operator\">&#061;<\/span>context_input_ids<span class=\"token punctuation\">,<\/span><br \/>\n            attention_mask<span class=\"token operator\">&#061;<\/span>context_attention_mask<br \/>\n        <span class=\"token punctuation\">)<\/span><br \/>\n        encoder_output <span class=\"token operator\">&#061;<\/span> bert_output<span class=\"token punctuation\">.<\/span>last_hidden_state<\/p>\n<p>        <span class=\"token comment\"># \u89e3\u7801\u5668<\/span><br \/>\n        decoder_output<span class=\"token punctuation\">,<\/span> _ <span class=\"token operator\">&#061;<\/span> self<span class=\"token punctuation\">.<\/span>decoder<span class=\"token punctuation\">(<\/span>response_input_ids<span class=\"token punctuation\">,<\/span> <span class=\"token punctuation\">(<\/span>encoder_output<span class=\"token punctuation\">[<\/span><span class=\"token punctuation\">:<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token number\">0<\/span><span class=\"token punctuation\">:<\/span><span class=\"token number\">1<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token punctuation\">:<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">,<\/span> encoder_output<span class=\"token punctuation\">[<\/span><span class=\"token punctuation\">:<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token number\">0<\/span><span class=\"token punctuation\">:<\/span><span class=\"token number\">1<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token punctuation\">:<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">)<\/span><\/p>\n<p>        <span class=\"token comment\"># \u8f93\u51fa\u5c42<\/span><br \/>\n        output <span class=\"token operator\">&#061;<\/span> self<span class=\"token punctuation\">.<\/span>fc<span class=\"token punctuation\">(<\/span>decoder_output<span class=\"token punctuation\">)<\/span><\/p>\n<p>        <span class=\"token keyword\">return<\/span> output<\/p>\n<p><span class=\"token comment\"># \u52a0\u8f7d\u6570\u636e<\/span><br \/>\ndata <span class=\"token operator\">&#061;<\/span> torch<span class=\"token punctuation\">.<\/span>load<span class=\"token punctuation\">(<\/span><span class=\"token string\">&#039;processed_data.pt&#039;<\/span><span class=\"token punctuation\">)<\/span><br \/>\ncontext_input_ids <span class=\"token operator\">&#061;<\/span> data<span class=\"token punctuation\">[<\/span><span class=\"token string\">&#039;context_input_ids&#039;<\/span><span class=\"token punctuation\">]<\/span><br \/>\ncontext_attention_mask <span class=\"token operator\">&#061;<\/span> data<span class=\"token punctuation\">[<\/span><span class=\"token string\">&#039;context_attention_mask&#039;<\/span><span class=\"token punctuation\">]<\/span><br \/>\nresponse_input_ids <span class=\"token operator\">&#061;<\/span> data<span class=\"token punctuation\">[<\/span><span class=\"token string\">&#039;response_input_ids&#039;<\/span><span class=\"token punctuation\">]<\/span><br \/>\nresponse_attention_mask <span class=\"token operator\">&#061;<\/span> data<span class=\"token punctuation\">[<\/span><span class=\"token string\">&#039;response_attention_mask&#039;<\/span><span class=\"token punctuation\">]<\/span><\/p>\n<p><span class=\"token comment\"># \u8d85\u53c2\u6570\u8bbe\u7f6e<\/span><br \/>\nbert_model_name <span class=\"token operator\">&#061;<\/span> <span class=\"token string\">&#039;bert-base-chinese&#039;<\/span><br \/>\nhidden_size <span class=\"token operator\">&#061;<\/span> <span class=\"token number\">768<\/span><br \/>\noutput_size <span class=\"token operator\">&#061;<\/span> tokenizer<span class=\"token punctuation\">.<\/span>vocab_size<br \/>\nbatch_size <span class=\"token operator\">&#061;<\/span> <span class=\"token number\">16<\/span><br \/>\nepochs <span class=\"token operator\">&#061;<\/span> <span class=\"token number\">10<\/span><br \/>\nlr <span class=\"token operator\">&#061;<\/span> <span class=\"token number\">0.0001<\/span><\/p>\n<p><span class=\"token comment\"># \u6a21\u578b\u5b9e\u4f8b\u5316<\/span><br \/>\nmodel <span class=\"token operator\">&#061;<\/span> BERTSeq2Seq<span class=\"token punctuation\">(<\/span>bert_model_name<span class=\"token punctuation\">,<\/span> hidden_size<span class=\"token punctuation\">,<\/span> output_size<span class=\"token punctuation\">)<\/span><\/p>\n<p><span class=\"token comment\"># \u635f\u5931\u51fd\u6570\u548c\u4f18\u5316\u5668<\/span><br \/>\ncriterion <span class=\"token operator\">&#061;<\/span> nn<span class=\"token punctuation\">.<\/span>CrossEntropyLoss<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><br \/>\noptimizer <span class=\"token operator\">&#061;<\/span> optim<span class=\"token punctuation\">.<\/span>Adam<span class=\"token punctuation\">(<\/span>model<span class=\"token punctuation\">.<\/span>parameters<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">,<\/span> lr<span class=\"token operator\">&#061;<\/span>lr<span class=\"token punctuation\">)<\/span><\/p>\n<p><span class=\"token comment\"># \u8bad\u7ec3\u8fc7\u7a0b<\/span><br \/>\nmodel<span class=\"token punctuation\">.<\/span>train<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><br \/>\n<span class=\"token keyword\">for<\/span> epoch <span class=\"token keyword\">in<\/span> <span class=\"token builtin\">range<\/span><span class=\"token punctuation\">(<\/span>epochs<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span><br \/>\n    total_loss <span class=\"token operator\">&#061;<\/span> <span class=\"token number\">0.0<\/span><br \/>\n    <span class=\"token keyword\">for<\/span> i <span class=\"token keyword\">in<\/span> <span class=\"token builtin\">range<\/span><span class=\"token punctuation\">(<\/span><span class=\"token number\">0<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token builtin\">len<\/span><span class=\"token punctuation\">(<\/span>context_input_ids<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">,<\/span> batch_size<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span><br \/>\n        optimizer<span class=\"token punctuation\">.<\/span>zero_grad<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><\/p>\n<p>        <span class=\"token comment\"># \u6279\u91cf\u52a0\u8f7d\u6570\u636e<\/span><br \/>\n        batch_context_input_ids <span class=\"token operator\">&#061;<\/span> context_input_ids<span class=\"token punctuation\">[<\/span>i<span class=\"token punctuation\">:<\/span>i<span class=\"token operator\">&#043;<\/span>batch_size<span class=\"token punctuation\">]<\/span><br \/>\n        batch_context_attention_mask <span class=\"token operator\">&#061;<\/span> context_attention_mask<span class=\"token punctuation\">[<\/span>i<span class=\"token punctuation\">:<\/span>i<span class=\"token operator\">&#043;<\/span>batch_size<span class=\"token punctuation\">]<\/span><br \/>\n        batch_response_input_ids <span class=\"token operator\">&#061;<\/span> response_input_ids<span class=\"token punctuation\">[<\/span>i<span class=\"token punctuation\">:<\/span>i<span class=\"token operator\">&#043;<\/span>batch_size<span class=\"token punctuation\">]<\/span><br \/>\n        batch_response_attention_mask <span class=\"token operator\">&#061;<\/span> response_attention_mask<span class=\"token punctuation\">[<\/span>i<span class=\"token punctuation\">:<\/span>i<span class=\"token operator\">&#043;<\/span>batch_size<span class=\"token punctuation\">]<\/span><\/p>\n<p>        <span class=\"token comment\"># \u6a21\u578b\u63a8\u7406<\/span><br \/>\n        output <span class=\"token operator\">&#061;<\/span> model<span class=\"token punctuation\">(<\/span><br \/>\n            batch_context_input_ids<span class=\"token punctuation\">,<\/span><br \/>\n            batch_context_attention_mask<span class=\"token punctuation\">,<\/span><br \/>\n            batch_response_input_ids<span class=\"token punctuation\">[<\/span><span class=\"token punctuation\">:<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token punctuation\">:<\/span><span class=\"token operator\">&#8211;<\/span><span class=\"token number\">1<\/span><span class=\"token punctuation\">]<\/span><br \/>\n        <span class=\"token punctuation\">)<\/span><\/p>\n<p>        <span class=\"token comment\"># \u8ba1\u7b97\u635f\u5931<\/span><br \/>\n        loss <span class=\"token operator\">&#061;<\/span> criterion<span class=\"token punctuation\">(<\/span><br \/>\n            output<span class=\"token punctuation\">.<\/span>reshape<span class=\"token punctuation\">(<\/span><span class=\"token operator\">&#8211;<\/span><span class=\"token number\">1<\/span><span class=\"token punctuation\">,<\/span> output<span class=\"token punctuation\">.<\/span>size<span class=\"token punctuation\">(<\/span><span class=\"token operator\">&#8211;<\/span><span class=\"token number\">1<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">,<\/span><br \/>\n            batch_response_input_ids<span class=\"token punctuation\">[<\/span><span class=\"token punctuation\">:<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token number\">1<\/span><span class=\"token punctuation\">:<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">.<\/span>reshape<span class=\"token punctuation\">(<\/span><span class=\"token operator\">&#8211;<\/span><span class=\"token number\">1<\/span><span class=\"token punctuation\">)<\/span><br \/>\n        <span class=\"token punctuation\">)<\/span><\/p>\n<p>        <span class=\"token comment\"># \u53cd\u5411\u4f20\u64ad\u548c\u4f18\u5316<\/span><br \/>\n        loss<span class=\"token punctuation\">.<\/span>backward<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><br \/>\n        optimizer<span class=\"token punctuation\">.<\/span>step<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><\/p>\n<p>        total_loss <span class=\"token operator\">&#043;&#061;<\/span> loss<span class=\"token punctuation\">.<\/span>item<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><\/p>\n<p>    average_loss <span class=\"token operator\">&#061;<\/span> total_loss <span class=\"token operator\">\/<\/span> <span class=\"token punctuation\">(<\/span><span class=\"token builtin\">len<\/span><span class=\"token punctuation\">(<\/span>context_input_ids<span class=\"token punctuation\">)<\/span> <span class=\"token operator\">\/\/<\/span> batch_size<span class=\"token punctuation\">)<\/span><br \/>\n    <span class=\"token keyword\">print<\/span><span class=\"token punctuation\">(<\/span><span class=\"token string-interpolation\"><span class=\"token string\">f&#034;Epoch: <\/span><span class=\"token interpolation\"><span class=\"token punctuation\">{<\/span>epoch <span class=\"token operator\">&#043;<\/span> <span class=\"token number\">1<\/span><span class=\"token punctuation\">}<\/span><\/span><span class=\"token string\">, Average Loss: <\/span><span class=\"token interpolation\"><span class=\"token punctuation\">{<\/span>average_loss<span class=\"token punctuation\">:<\/span><span class=\"token format-spec\">.4f<\/span><span class=\"token punctuation\">}<\/span><\/span><span class=\"token string\">&#034;<\/span><\/span><span class=\"token punctuation\">)<\/span><\/p>\n<p><span class=\"token comment\"># \u4fdd\u5b58\u6a21\u578b<\/span><br \/>\ntorch<span class=\"token punctuation\">.<\/span>save<span class=\"token punctuation\">(<\/span>model<span class=\"token punctuation\">.<\/span>state_dict<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token string\">&#039;bert_seq2seq_model.pt&#039;<\/span><span class=\"token punctuation\">)<\/span><\/p>\n<hr \/>\n<h3>\u516d\u3001\u6a21\u578b\u90e8\u7f72\u4e0e\u6d4b\u8bd5<\/h3>\n<h4>6.1 \u6a21\u578b\u90e8\u7f72\u65b9\u5f0f<\/h4>\n<p>&#x1f4a1; \u6a21\u578b\u90e8\u7f72\u7684\u65b9\u5f0f\u6709\u591a\u79cd&#xff0c;\u5e38\u89c1\u7684\u5305\u62ec&#xff1a;<\/p>\n<ul>\n<li>\u672c\u5730\u90e8\u7f72&#xff1a;\u5c06\u6a21\u578b\u90e8\u7f72\u5728\u672c\u5730\u670d\u52a1\u5668\u4e0a&#xff0c;\u901a\u8fc7API\u63d0\u4f9b\u670d\u52a1\u3002<\/li>\n<li>\u4e91\u7aef\u90e8\u7f72&#xff1a;\u5c06\u6a21\u578b\u90e8\u7f72\u5728\u4e91\u8ba1\u7b97\u5e73\u53f0\u4e0a&#xff0c;\u5982AWS\u3001Azure\u3001\u963f\u91cc\u4e91\u7b49\u3002<\/li>\n<li>\u8fb9\u7f18\u90e8\u7f72&#xff1a;\u5c06\u6a21\u578b\u90e8\u7f72\u5728\u8fb9\u7f18\u8bbe\u5907\u4e0a&#xff0c;\u5982\u624b\u673a\u3001\u667a\u80fd\u97f3\u7bb1\u7b49\u3002<\/li>\n<\/ul>\n<h4>6.2 \u6a21\u578b\u6d4b\u8bd5\u65b9\u6cd5<\/h4>\n<p>\u26a0\ufe0f \u6a21\u578b\u6d4b\u8bd5\u4e3b\u8981\u5305\u62ec\u4ee5\u4e0b\u5185\u5bb9&#xff1a;<\/p>\n<li>\u81ea\u52a8\u8bc4\u4f30&#xff1a;\u4f7f\u7528\u81ea\u52a8\u8bc4\u4f30\u6307\u6807&#xff0c;\u5982BLEU\u3001ROUGE\u7b49&#xff0c;\u8bc4\u4f30\u6a21\u578b\u7684\u8f93\u51fa\u8d28\u91cf\u3002<\/li>\n<li>\u4eba\u5de5\u8bc4\u4f30&#xff1a;\u9080\u8bf7\u7528\u6237\u5bf9\u6a21\u578b\u7684\u8f93\u51fa\u8fdb\u884c\u8bc4\u4f30&#xff0c;\u4ee5\u83b7\u5f97\u66f4\u771f\u5b9e\u7684\u53cd\u9988\u3002<\/li>\n<li>\u538b\u529b\u6d4b\u8bd5&#xff1a;\u6d4b\u8bd5\u6a21\u578b\u5728\u9ad8\u5e76\u53d1\u60c5\u51b5\u4e0b\u7684\u6027\u80fd\u548c\u7a33\u5b9a\u6027\u3002<\/li>\n<h4>6.3 \u4ee3\u7801\u5b9e\u73b0&#xff1a;\u6a21\u578b\u90e8\u7f72\u4e0e\u6d4b\u8bd5<\/h4>\n<p><span class=\"token keyword\">import<\/span> torch<br \/>\n<span class=\"token keyword\">from<\/span> transformers <span class=\"token keyword\">import<\/span> BertTokenizer<br \/>\n<span class=\"token keyword\">from<\/span> flask <span class=\"token keyword\">import<\/span> Flask<span class=\"token punctuation\">,<\/span> request<span class=\"token punctuation\">,<\/span> jsonify<\/p>\n<p><span class=\"token comment\"># \u52a0\u8f7d\u6a21\u578b\u548c\u5206\u8bcd\u5668<\/span><br \/>\ntokenizer <span class=\"token operator\">&#061;<\/span> BertTokenizer<span class=\"token punctuation\">.<\/span>from_pretrained<span class=\"token punctuation\">(<\/span><span class=\"token string\">&#039;bert-base-chinese&#039;<\/span><span class=\"token punctuation\">)<\/span><br \/>\nmodel <span class=\"token operator\">&#061;<\/span> BERTSeq2Seq<span class=\"token punctuation\">(<\/span><span class=\"token string\">&#039;bert-base-chinese&#039;<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token number\">768<\/span><span class=\"token punctuation\">,<\/span> tokenizer<span class=\"token punctuation\">.<\/span>vocab_size<span class=\"token punctuation\">)<\/span><br \/>\nmodel<span class=\"token punctuation\">.<\/span>load_state_dict<span class=\"token punctuation\">(<\/span>torch<span class=\"token punctuation\">.<\/span>load<span class=\"token punctuation\">(<\/span><span class=\"token string\">&#039;bert_seq2seq_model.pt&#039;<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">)<\/span><br \/>\nmodel<span class=\"token punctuation\">.<\/span><span class=\"token builtin\">eval<\/span><span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><\/p>\n<p><span class=\"token comment\"># \u521d\u59cb\u5316Flask\u5e94\u7528<\/span><br \/>\napp <span class=\"token operator\">&#061;<\/span> Flask<span class=\"token punctuation\">(<\/span>__name__<span class=\"token punctuation\">)<\/span><\/p>\n<p><span class=\"token comment\"># \u5b9a\u4e49API\u63a5\u53e3<\/span><br \/>\n<span class=\"token decorator annotation punctuation\">&#064;app<span class=\"token punctuation\">.<\/span>route<\/span><span class=\"token punctuation\">(<\/span><span class=\"token string\">&#039;\/chat&#039;<\/span><span class=\"token punctuation\">,<\/span> methods<span class=\"token operator\">&#061;<\/span><span class=\"token punctuation\">[<\/span><span class=\"token string\">&#039;POST&#039;<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">)<\/span><br \/>\n<span class=\"token keyword\">def<\/span> <span class=\"token function\">chat<\/span><span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span><br \/>\n    <span class=\"token comment\"># \u83b7\u53d6\u7528\u6237\u8f93\u5165<\/span><br \/>\n    data <span class=\"token operator\">&#061;<\/span> request<span class=\"token punctuation\">.<\/span>get_json<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><br \/>\n    context <span class=\"token operator\">&#061;<\/span> data<span class=\"token punctuation\">[<\/span><span class=\"token string\">&#039;context&#039;<\/span><span class=\"token punctuation\">]<\/span><\/p>\n<p>    <span class=\"token comment\"># \u6587\u672c\u9884\u5904\u7406<\/span><br \/>\n    context_encoding <span class=\"token operator\">&#061;<\/span> tokenizer<span class=\"token punctuation\">(<\/span><br \/>\n        context<span class=\"token punctuation\">,<\/span><br \/>\n        padding<span class=\"token operator\">&#061;<\/span><span class=\"token string\">&#039;max_length&#039;<\/span><span class=\"token punctuation\">,<\/span><br \/>\n        truncation<span class=\"token operator\">&#061;<\/span><span class=\"token boolean\">True<\/span><span class=\"token punctuation\">,<\/span><br \/>\n        max_length<span class=\"token operator\">&#061;<\/span><span class=\"token number\">512<\/span><span class=\"token punctuation\">,<\/span><br \/>\n        return_tensors<span class=\"token operator\">&#061;<\/span><span class=\"token string\">&#039;pt&#039;<\/span><br \/>\n    <span class=\"token punctuation\">)<\/span><\/p>\n<p>    <span class=\"token comment\"># \u6a21\u578b\u63a8\u7406<\/span><br \/>\n    <span class=\"token keyword\">with<\/span> torch<span class=\"token punctuation\">.<\/span>no_grad<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span><br \/>\n        <span class=\"token comment\"># \u751f\u6210\u56de\u590d<\/span><br \/>\n        response <span class=\"token operator\">&#061;<\/span> generate_response<span class=\"token punctuation\">(<\/span>model<span class=\"token punctuation\">,<\/span> context_encoding<span class=\"token punctuation\">,<\/span> tokenizer<span class=\"token punctuation\">)<\/span><\/p>\n<p>    <span class=\"token comment\"># \u8fd4\u56de\u56de\u590d<\/span><br \/>\n    <span class=\"token keyword\">return<\/span> jsonify<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">{<\/span><span class=\"token string\">&#039;response&#039;<\/span><span class=\"token punctuation\">:<\/span> response<span class=\"token punctuation\">}<\/span><span class=\"token punctuation\">)<\/span><\/p>\n<p><span class=\"token comment\"># \u56de\u590d\u751f\u6210\u51fd\u6570<\/span><br \/>\n<span class=\"token keyword\">def<\/span> <span class=\"token function\">generate_response<\/span><span class=\"token punctuation\">(<\/span>model<span class=\"token punctuation\">,<\/span> context_encoding<span class=\"token punctuation\">,<\/span> tokenizer<span class=\"token punctuation\">,<\/span> max_length<span class=\"token operator\">&#061;<\/span><span class=\"token number\">512<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span><br \/>\n    <span class=\"token comment\"># \u521d\u59cb\u5316\u56de\u590d\u5e8f\u5217<\/span><br \/>\n    response_input_ids <span class=\"token operator\">&#061;<\/span> torch<span class=\"token punctuation\">.<\/span>tensor<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">[<\/span><span class=\"token punctuation\">[<\/span>tokenizer<span class=\"token punctuation\">.<\/span>cls_token_id<span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">)<\/span><\/p>\n<p>    <span class=\"token comment\"># \u9010\u8bcd\u751f\u6210\u56de\u590d<\/span><br \/>\n    <span class=\"token keyword\">for<\/span> _ <span class=\"token keyword\">in<\/span> <span class=\"token builtin\">range<\/span><span class=\"token punctuation\">(<\/span>max_length<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span><br \/>\n        <span class=\"token comment\"># \u6a21\u578b\u63a8\u7406<\/span><br \/>\n        output <span class=\"token operator\">&#061;<\/span> model<span class=\"token punctuation\">(<\/span><br \/>\n            context_encoding<span class=\"token punctuation\">[<\/span><span class=\"token string\">&#039;input_ids&#039;<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">,<\/span><br \/>\n            context_encoding<span class=\"token punctuation\">[<\/span><span class=\"token string\">&#039;attention_mask&#039;<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">,<\/span><br \/>\n            response_input_ids<br \/>\n        <span class=\"token punctuation\">)<\/span><\/p>\n<p>        <span class=\"token comment\"># \u83b7\u53d6\u4e0b\u4e00\u4e2a\u8bcd\u7684\u6982\u7387<\/span><br \/>\n        next_token_logits <span class=\"token operator\">&#061;<\/span> output<span class=\"token punctuation\">[<\/span><span class=\"token punctuation\">:<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token operator\">&#8211;<\/span><span class=\"token number\">1<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token punctuation\">:<\/span><span class=\"token punctuation\">]<\/span><br \/>\n        next_token_id <span class=\"token operator\">&#061;<\/span> torch<span class=\"token punctuation\">.<\/span>argmax<span class=\"token punctuation\">(<\/span>next_token_logits<span class=\"token punctuation\">,<\/span> dim<span class=\"token operator\">&#061;<\/span><span class=\"token operator\">&#8211;<\/span><span class=\"token number\">1<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">.<\/span>unsqueeze<span class=\"token punctuation\">(<\/span><span class=\"token number\">1<\/span><span class=\"token punctuation\">)<\/span><\/p>\n<p>        <span class=\"token comment\"># \u6dfb\u52a0\u5230\u56de\u590d\u5e8f\u5217<\/span><br \/>\n        response_input_ids <span class=\"token operator\">&#061;<\/span> torch<span class=\"token punctuation\">.<\/span>cat<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">[<\/span>response_input_ids<span class=\"token punctuation\">,<\/span> next_token_id<span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">,<\/span> dim<span class=\"token operator\">&#061;<\/span><span class=\"token number\">1<\/span><span class=\"token punctuation\">)<\/span><\/p>\n<p>        <span class=\"token comment\"># \u68c0\u67e5\u662f\u5426\u751f\u6210\u4e86\u7ed3\u675f\u7b26<\/span><br \/>\n        <span class=\"token keyword\">if<\/span> next_token_id<span class=\"token punctuation\">.<\/span>item<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span> <span class=\"token operator\">&#061;&#061;<\/span> tokenizer<span class=\"token punctuation\">.<\/span>sep_token_id<span class=\"token punctuation\">:<\/span><br \/>\n            <span class=\"token keyword\">break<\/span><\/p>\n<p>    <span class=\"token comment\"># \u89e3\u7801\u56de\u590d<\/span><br \/>\n    response <span class=\"token operator\">&#061;<\/span> tokenizer<span class=\"token punctuation\">.<\/span>decode<span class=\"token punctuation\">(<\/span>response_input_ids<span class=\"token punctuation\">.<\/span>squeeze<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">,<\/span> skip_special_tokens<span class=\"token operator\">&#061;<\/span><span class=\"token boolean\">True<\/span><span class=\"token punctuation\">)<\/span><\/p>\n<p>    <span class=\"token keyword\">return<\/span> response<\/p>\n<p><span class=\"token comment\"># \u8fd0\u884cFlask\u5e94\u7528<\/span><br \/>\n<span class=\"token keyword\">if<\/span> __name__ <span class=\"token operator\">&#061;&#061;<\/span> <span class=\"token string\">&#039;__main__&#039;<\/span><span class=\"token punctuation\">:<\/span><br \/>\n    app<span class=\"token punctuation\">.<\/span>run<span class=\"token punctuation\">(<\/span>host<span class=\"token operator\">&#061;<\/span><span class=\"token string\">&#039;0.0.0.0&#039;<\/span><span class=\"token punctuation\">,<\/span> port<span class=\"token operator\">&#061;<\/span><span class=\"token number\">5000<\/span><span class=\"token punctuation\">)<\/span><\/p>\n<hr \/>\n<h3>\u4e03\u3001\u6848\u4f8b\u5206\u6790\u4e0e\u4f18\u5316\u601d\u8def<\/h3>\n<h4>7.1 \u6848\u4f8b\u5206\u6790<\/h4>\n<p>\u2705 \u6211\u4eec\u4f7f\u7528DailyDialog\u6570\u636e\u96c6\u6784\u5efa\u4e86\u4e00\u4e2a\u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf\u3002DailyDialog\u6570\u636e\u96c6\u5305\u542b13118\u4e2a\u5bf9\u8bdd&#xff0c;\u6bcf\u4e2a\u5bf9\u8bdd\u5305\u542b3-10\u8f6e&#xff0c;\u6db5\u76d6\u4e86\u65e5\u5e38\u751f\u6d3b\u4e2d\u7684\u5404\u79cd\u573a\u666f\u3002<\/p>\n<h4>7.2 \u6a21\u578b\u6027\u80fd\u8bc4\u4f30<\/h4>\n<p>\u26a0\ufe0f \u6211\u4eec\u4f7f\u7528BLEU\u3001ROUGE\u548cMETEOR\u7b49\u6307\u6807\u8bc4\u4f30\u4e86\u6a21\u578b\u7684\u6027\u80fd\u3002\u7ed3\u679c\u8868\u660e&#xff0c;\u6211\u4eec\u7684\u6a21\u578b\u5728DailyDialog\u6570\u636e\u96c6\u4e0a\u53d6\u5f97\u4e86\u8f83\u597d\u7684\u6027\u80fd\u3002<\/p>\n<h4>7.3 \u4f18\u5316\u601d\u8def<\/h4>\n<p>&#x1f4a1; \u4ee5\u4e0b\u662f\u4e00\u4e9b\u4f18\u5316\u6a21\u578b\u6027\u80fd\u7684\u601d\u8def&#xff1a;<\/p>\n<li>\u4f7f\u7528\u66f4\u5927\u7684\u9884\u8bad\u7ec3\u6a21\u578b&#xff1a;\u5982BERT-large\u3001RoBERTa\u7b49\u3002<\/li>\n<li>\u589e\u52a0\u8bad\u7ec3\u6570\u636e&#xff1a;\u4f7f\u7528\u66f4\u591a\u7684\u5bf9\u8bdd\u6570\u636e\u8fdb\u884c\u8bad\u7ec3\u3002<\/li>\n<li>\u4f18\u5316\u6a21\u578b\u67b6\u6784&#xff1a;\u5982\u4f7f\u7528TransformerDecoder\u66ff\u4ee3LSTMDecoder\u3002<\/li>\n<li>\u8c03\u6574\u8d85\u53c2\u6570&#xff1a;\u5982\u5b66\u4e60\u7387\u3001\u6279\u6b21\u5927\u5c0f\u3001\u6700\u5927\u5e8f\u5217\u957f\u5ea6\u7b49\u3002<\/li>\n<li>\u6dfb\u52a0\u6ce8\u610f\u529b\u673a\u5236&#xff1a;\u5728\u89e3\u7801\u5668\u4e2d\u6dfb\u52a0\u6ce8\u610f\u529b\u673a\u5236&#xff0c;\u4ee5\u63d0\u9ad8\u8f93\u51fa\u8d28\u91cf\u3002<\/li>\n<hr \/>\n<h3>\u516b\u3001\u603b\u7ed3<\/h3>\n<p>&#x1f4a1; \u672c\u6587\u8be6\u7ec6\u4ecb\u7ecd\u4e86\u5982\u4f55\u4f7f\u7528BERT&#043;Seq2Seq\u67b6\u6784\u6784\u5efa\u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf\u3002\u6211\u4eec\u9996\u5148\u4ecb\u7ecd\u4e86\u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf\u7684\u6838\u5fc3\u539f\u7406\u4e0e\u67b6\u6784&#xff0c;\u7136\u540e\u8bb2\u89e3\u4e86BERT\u6a21\u578b\u548cSeq2Seq\u6a21\u578b\u7684\u539f\u7406\u4e0e\u5e94\u7528&#xff0c;\u63a5\u7740\u4ecb\u7ecd\u4e86\u6570\u636e\u96c6\u51c6\u5907\u4e0e\u9884\u5904\u7406\u3001\u6a21\u578b\u8bad\u7ec3\u4e0e\u4f18\u5316\u3001\u6a21\u578b\u90e8\u7f72\u4e0e\u6d4b\u8bd5\u7b49\u6b65\u9aa4&#xff0c;\u6700\u540e\u901a\u8fc7\u6848\u4f8b\u5206\u6790\u548c\u4f18\u5316\u601d\u8def\u8fdb\u884c\u4e86\u603b\u7ed3\u3002<\/p>\n<p>\u2705 \u5e0c\u671b\u672c\u6587\u80fd\u591f\u5e2e\u52a9\u8bfb\u8005\u7406\u89e3\u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf\u7684\u6838\u5fc3\u6280\u672f&#xff0c;\u5e76\u80fd\u591f\u72ec\u7acb\u5b8c\u6210\u4e00\u4e2a\u57fa\u4e8eBERT&#043;Seq2Seq\u67b6\u6784\u7684\u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf\u3002<\/p>\n","protected":false},"excerpt":{"rendered":"<p>\u7b2c8\u7bc7&#xff1a;\u81ea\u7136\u8bed\u8a00\u5904\u7406\u5b9e\u6218\u2014\u2014\u6784\u5efa\u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf&#xff08;BERTSeq2Seq\u67b6\u6784&#xff09;\u5b66\u4e60\u76ee\u6807<br \/>\n\u7406\u89e3\u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf\u7684\u6838\u5fc3\u539f\u7406\u4e0e\u67b6\u6784\u638c\u63e1BERT\u6a21\u578b\u5728\u6587\u672c\u7406\u89e3\u4e2d\u7684\u5e94\u7528\u65b9\u6cd5\u5b66\u4f1a\u4f7f\u7528Seq2Seq\u6a21\u578b\u5b9e\u73b0\u6587\u672c\u751f\u6210\u529f\u80fd\u7406\u89e3Attention\u673a\u5236\u5728\u5bf9\u8bdd\u7cfb\u7edf\u4e2d\u7684\u91cd\u8981\u6027\u80fd\u591f\u72ec\u7acb\u5b8c\u6210\u4e00\u4e2a\u57fa\u4e8eBERTSeq2Seq\u67b6\u6784\u7684\u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf<br \/>\n\u7ae0\u8282\u91cd\u70b9<br \/>\n\u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf\u6982\u8ff0BERT\u6a21\u578b\u539f\u7406\u4e0e\u5e94\u7528Seq2Seq\u6a21\u578b\u4e0eAttention\u673a\u5236\u6570\u636e\u96c6\u51c6\u5907\u4e0e\u9884\u5904\u7406\u6a21\u578b\u8bad\u7ec3\u4e0e\u4f18\u5316\u6a21\u578b\u90e8\u7f72\u4e0e\u6d4b<\/p>\n","protected":false},"author":2,"featured_media":60872,"comment_status":"open","ping_status":"open","sticky":false,"template":"","format":"standard","meta":{"footnotes":""},"categories":[1],"tags":[2554,188,224],"topic":[],"class_list":["post-60873","post","type-post","status-publish","format-standard","has-post-thumbnail","hentry","category-server","tag-bert","tag-188","tag-224"],"yoast_head":"<!-- This site is optimized with the Yoast SEO plugin v20.3 - https:\/\/yoast.com\/wordpress\/plugins\/seo\/ -->\n<title>\u81ea\u7136\u8bed\u8a00\u5904\u7406\u5b9e\u6218\u2014\u2014\u6784\u5efa\u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf\uff08BERT+Seq2Seq\u67b6\u6784\uff09 - \u7f51\u7855\u4e92\u8054\u5e2e\u52a9\u4e2d\u5fc3<\/title>\n<meta name=\"robots\" content=\"index, follow, max-snippet:-1, max-image-preview:large, max-video-preview:-1\" \/>\n<link rel=\"canonical\" href=\"https:\/\/www.wsisp.com\/helps\/60873.html\" \/>\n<meta property=\"og:locale\" content=\"zh_CN\" \/>\n<meta property=\"og:type\" content=\"article\" \/>\n<meta property=\"og:title\" content=\"\u81ea\u7136\u8bed\u8a00\u5904\u7406\u5b9e\u6218\u2014\u2014\u6784\u5efa\u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf\uff08BERT+Seq2Seq\u67b6\u6784\uff09 - \u7f51\u7855\u4e92\u8054\u5e2e\u52a9\u4e2d\u5fc3\" \/>\n<meta property=\"og:description\" content=\"\u7b2c8\u7bc7&#xff1a;\u81ea\u7136\u8bed\u8a00\u5904\u7406\u5b9e\u6218\u2014\u2014\u6784\u5efa\u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf&#xff08;BERTSeq2Seq\u67b6\u6784&#xff09;\u5b66\u4e60\u76ee\u6807 \u7406\u89e3\u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf\u7684\u6838\u5fc3\u539f\u7406\u4e0e\u67b6\u6784\u638c\u63e1BERT\u6a21\u578b\u5728\u6587\u672c\u7406\u89e3\u4e2d\u7684\u5e94\u7528\u65b9\u6cd5\u5b66\u4f1a\u4f7f\u7528Seq2Seq\u6a21\u578b\u5b9e\u73b0\u6587\u672c\u751f\u6210\u529f\u80fd\u7406\u89e3Attention\u673a\u5236\u5728\u5bf9\u8bdd\u7cfb\u7edf\u4e2d\u7684\u91cd\u8981\u6027\u80fd\u591f\u72ec\u7acb\u5b8c\u6210\u4e00\u4e2a\u57fa\u4e8eBERTSeq2Seq\u67b6\u6784\u7684\u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf \u7ae0\u8282\u91cd\u70b9 \u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf\u6982\u8ff0BERT\u6a21\u578b\u539f\u7406\u4e0e\u5e94\u7528Seq2Seq\u6a21\u578b\u4e0eAttention\u673a\u5236\u6570\u636e\u96c6\u51c6\u5907\u4e0e\u9884\u5904\u7406\u6a21\u578b\u8bad\u7ec3\u4e0e\u4f18\u5316\u6a21\u578b\u90e8\u7f72\u4e0e\u6d4b\" \/>\n<meta property=\"og:url\" content=\"https:\/\/www.wsisp.com\/helps\/60873.html\" \/>\n<meta property=\"og:site_name\" content=\"\u7f51\u7855\u4e92\u8054\u5e2e\u52a9\u4e2d\u5fc3\" \/>\n<meta property=\"article:published_time\" content=\"2026-01-16T06:17:33+00:00\" \/>\n<meta property=\"og:image\" content=\"https:\/\/www.wsisp.com\/helps\/wp-content\/uploads\/2026\/01\/20260116061731-6969d7fbcbf1c.png\" \/>\n<meta name=\"author\" content=\"admin\" \/>\n<meta name=\"twitter:card\" content=\"summary_large_image\" \/>\n<meta name=\"twitter:label1\" content=\"\u4f5c\u8005\" \/>\n\t<meta name=\"twitter:data1\" content=\"admin\" \/>\n\t<meta name=\"twitter:label2\" content=\"\u9884\u8ba1\u9605\u8bfb\u65f6\u95f4\" \/>\n\t<meta name=\"twitter:data2\" content=\"6 \u5206\" \/>\n<script type=\"application\/ld+json\" class=\"yoast-schema-graph\">{\"@context\":\"https:\/\/schema.org\",\"@graph\":[{\"@type\":\"WebPage\",\"@id\":\"https:\/\/www.wsisp.com\/helps\/60873.html\",\"url\":\"https:\/\/www.wsisp.com\/helps\/60873.html\",\"name\":\"\u81ea\u7136\u8bed\u8a00\u5904\u7406\u5b9e\u6218\u2014\u2014\u6784\u5efa\u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf\uff08BERT+Seq2Seq\u67b6\u6784\uff09 - \u7f51\u7855\u4e92\u8054\u5e2e\u52a9\u4e2d\u5fc3\",\"isPartOf\":{\"@id\":\"https:\/\/www.wsisp.com\/helps\/#website\"},\"datePublished\":\"2026-01-16T06:17:33+00:00\",\"dateModified\":\"2026-01-16T06:17:33+00:00\",\"author\":{\"@id\":\"https:\/\/www.wsisp.com\/helps\/#\/schema\/person\/358e386c577a3ab51c4493330a20ad41\"},\"breadcrumb\":{\"@id\":\"https:\/\/www.wsisp.com\/helps\/60873.html#breadcrumb\"},\"inLanguage\":\"zh-Hans\",\"potentialAction\":[{\"@type\":\"ReadAction\",\"target\":[\"https:\/\/www.wsisp.com\/helps\/60873.html\"]}]},{\"@type\":\"BreadcrumbList\",\"@id\":\"https:\/\/www.wsisp.com\/helps\/60873.html#breadcrumb\",\"itemListElement\":[{\"@type\":\"ListItem\",\"position\":1,\"name\":\"\u9996\u9875\",\"item\":\"https:\/\/www.wsisp.com\/helps\"},{\"@type\":\"ListItem\",\"position\":2,\"name\":\"\u81ea\u7136\u8bed\u8a00\u5904\u7406\u5b9e\u6218\u2014\u2014\u6784\u5efa\u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf\uff08BERT+Seq2Seq\u67b6\u6784\uff09\"}]},{\"@type\":\"WebSite\",\"@id\":\"https:\/\/www.wsisp.com\/helps\/#website\",\"url\":\"https:\/\/www.wsisp.com\/helps\/\",\"name\":\"\u7f51\u7855\u4e92\u8054\u5e2e\u52a9\u4e2d\u5fc3\",\"description\":\"\u9999\u6e2f\u670d\u52a1\u5668_\u9999\u6e2f\u4e91\u670d\u52a1\u5668\u8d44\u8baf_\u670d\u52a1\u5668\u5e2e\u52a9\u6587\u6863_\u670d\u52a1\u5668\u6559\u7a0b\",\"potentialAction\":[{\"@type\":\"SearchAction\",\"target\":{\"@type\":\"EntryPoint\",\"urlTemplate\":\"https:\/\/www.wsisp.com\/helps\/?s={search_term_string}\"},\"query-input\":\"required name=search_term_string\"}],\"inLanguage\":\"zh-Hans\"},{\"@type\":\"Person\",\"@id\":\"https:\/\/www.wsisp.com\/helps\/#\/schema\/person\/358e386c577a3ab51c4493330a20ad41\",\"name\":\"admin\",\"image\":{\"@type\":\"ImageObject\",\"inLanguage\":\"zh-Hans\",\"@id\":\"https:\/\/www.wsisp.com\/helps\/#\/schema\/person\/image\/\",\"url\":\"https:\/\/gravatar.wp-china-yes.net\/avatar\/?s=96&d=mystery\",\"contentUrl\":\"https:\/\/gravatar.wp-china-yes.net\/avatar\/?s=96&d=mystery\",\"caption\":\"admin\"},\"sameAs\":[\"http:\/\/wp.wsisp.com\"],\"url\":\"https:\/\/www.wsisp.com\/helps\/author\/admin\"}]}<\/script>\n<!-- \/ Yoast SEO plugin. -->","yoast_head_json":{"title":"\u81ea\u7136\u8bed\u8a00\u5904\u7406\u5b9e\u6218\u2014\u2014\u6784\u5efa\u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf\uff08BERT+Seq2Seq\u67b6\u6784\uff09 - \u7f51\u7855\u4e92\u8054\u5e2e\u52a9\u4e2d\u5fc3","robots":{"index":"index","follow":"follow","max-snippet":"max-snippet:-1","max-image-preview":"max-image-preview:large","max-video-preview":"max-video-preview:-1"},"canonical":"https:\/\/www.wsisp.com\/helps\/60873.html","og_locale":"zh_CN","og_type":"article","og_title":"\u81ea\u7136\u8bed\u8a00\u5904\u7406\u5b9e\u6218\u2014\u2014\u6784\u5efa\u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf\uff08BERT+Seq2Seq\u67b6\u6784\uff09 - \u7f51\u7855\u4e92\u8054\u5e2e\u52a9\u4e2d\u5fc3","og_description":"\u7b2c8\u7bc7&#xff1a;\u81ea\u7136\u8bed\u8a00\u5904\u7406\u5b9e\u6218\u2014\u2014\u6784\u5efa\u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf&#xff08;BERTSeq2Seq\u67b6\u6784&#xff09;\u5b66\u4e60\u76ee\u6807 \u7406\u89e3\u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf\u7684\u6838\u5fc3\u539f\u7406\u4e0e\u67b6\u6784\u638c\u63e1BERT\u6a21\u578b\u5728\u6587\u672c\u7406\u89e3\u4e2d\u7684\u5e94\u7528\u65b9\u6cd5\u5b66\u4f1a\u4f7f\u7528Seq2Seq\u6a21\u578b\u5b9e\u73b0\u6587\u672c\u751f\u6210\u529f\u80fd\u7406\u89e3Attention\u673a\u5236\u5728\u5bf9\u8bdd\u7cfb\u7edf\u4e2d\u7684\u91cd\u8981\u6027\u80fd\u591f\u72ec\u7acb\u5b8c\u6210\u4e00\u4e2a\u57fa\u4e8eBERTSeq2Seq\u67b6\u6784\u7684\u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf \u7ae0\u8282\u91cd\u70b9 \u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf\u6982\u8ff0BERT\u6a21\u578b\u539f\u7406\u4e0e\u5e94\u7528Seq2Seq\u6a21\u578b\u4e0eAttention\u673a\u5236\u6570\u636e\u96c6\u51c6\u5907\u4e0e\u9884\u5904\u7406\u6a21\u578b\u8bad\u7ec3\u4e0e\u4f18\u5316\u6a21\u578b\u90e8\u7f72\u4e0e\u6d4b","og_url":"https:\/\/www.wsisp.com\/helps\/60873.html","og_site_name":"\u7f51\u7855\u4e92\u8054\u5e2e\u52a9\u4e2d\u5fc3","article_published_time":"2026-01-16T06:17:33+00:00","og_image":[{"url":"https:\/\/www.wsisp.com\/helps\/wp-content\/uploads\/2026\/01\/20260116061731-6969d7fbcbf1c.png"}],"author":"admin","twitter_card":"summary_large_image","twitter_misc":{"\u4f5c\u8005":"admin","\u9884\u8ba1\u9605\u8bfb\u65f6\u95f4":"6 \u5206"},"schema":{"@context":"https:\/\/schema.org","@graph":[{"@type":"WebPage","@id":"https:\/\/www.wsisp.com\/helps\/60873.html","url":"https:\/\/www.wsisp.com\/helps\/60873.html","name":"\u81ea\u7136\u8bed\u8a00\u5904\u7406\u5b9e\u6218\u2014\u2014\u6784\u5efa\u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf\uff08BERT+Seq2Seq\u67b6\u6784\uff09 - \u7f51\u7855\u4e92\u8054\u5e2e\u52a9\u4e2d\u5fc3","isPartOf":{"@id":"https:\/\/www.wsisp.com\/helps\/#website"},"datePublished":"2026-01-16T06:17:33+00:00","dateModified":"2026-01-16T06:17:33+00:00","author":{"@id":"https:\/\/www.wsisp.com\/helps\/#\/schema\/person\/358e386c577a3ab51c4493330a20ad41"},"breadcrumb":{"@id":"https:\/\/www.wsisp.com\/helps\/60873.html#breadcrumb"},"inLanguage":"zh-Hans","potentialAction":[{"@type":"ReadAction","target":["https:\/\/www.wsisp.com\/helps\/60873.html"]}]},{"@type":"BreadcrumbList","@id":"https:\/\/www.wsisp.com\/helps\/60873.html#breadcrumb","itemListElement":[{"@type":"ListItem","position":1,"name":"\u9996\u9875","item":"https:\/\/www.wsisp.com\/helps"},{"@type":"ListItem","position":2,"name":"\u81ea\u7136\u8bed\u8a00\u5904\u7406\u5b9e\u6218\u2014\u2014\u6784\u5efa\u667a\u80fd\u5bf9\u8bdd\u7cfb\u7edf\uff08BERT+Seq2Seq\u67b6\u6784\uff09"}]},{"@type":"WebSite","@id":"https:\/\/www.wsisp.com\/helps\/#website","url":"https:\/\/www.wsisp.com\/helps\/","name":"\u7f51\u7855\u4e92\u8054\u5e2e\u52a9\u4e2d\u5fc3","description":"\u9999\u6e2f\u670d\u52a1\u5668_\u9999\u6e2f\u4e91\u670d\u52a1\u5668\u8d44\u8baf_\u670d\u52a1\u5668\u5e2e\u52a9\u6587\u6863_\u670d\u52a1\u5668\u6559\u7a0b","potentialAction":[{"@type":"SearchAction","target":{"@type":"EntryPoint","urlTemplate":"https:\/\/www.wsisp.com\/helps\/?s={search_term_string}"},"query-input":"required name=search_term_string"}],"inLanguage":"zh-Hans"},{"@type":"Person","@id":"https:\/\/www.wsisp.com\/helps\/#\/schema\/person\/358e386c577a3ab51c4493330a20ad41","name":"admin","image":{"@type":"ImageObject","inLanguage":"zh-Hans","@id":"https:\/\/www.wsisp.com\/helps\/#\/schema\/person\/image\/","url":"https:\/\/gravatar.wp-china-yes.net\/avatar\/?s=96&d=mystery","contentUrl":"https:\/\/gravatar.wp-china-yes.net\/avatar\/?s=96&d=mystery","caption":"admin"},"sameAs":["http:\/\/wp.wsisp.com"],"url":"https:\/\/www.wsisp.com\/helps\/author\/admin"}]}},"_links":{"self":[{"href":"https:\/\/www.wsisp.com\/helps\/wp-json\/wp\/v2\/posts\/60873","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/www.wsisp.com\/helps\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/www.wsisp.com\/helps\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/www.wsisp.com\/helps\/wp-json\/wp\/v2\/users\/2"}],"replies":[{"embeddable":true,"href":"https:\/\/www.wsisp.com\/helps\/wp-json\/wp\/v2\/comments?post=60873"}],"version-history":[{"count":0,"href":"https:\/\/www.wsisp.com\/helps\/wp-json\/wp\/v2\/posts\/60873\/revisions"}],"wp:featuredmedia":[{"embeddable":true,"href":"https:\/\/www.wsisp.com\/helps\/wp-json\/wp\/v2\/media\/60872"}],"wp:attachment":[{"href":"https:\/\/www.wsisp.com\/helps\/wp-json\/wp\/v2\/media?parent=60873"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/www.wsisp.com\/helps\/wp-json\/wp\/v2\/categories?post=60873"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/www.wsisp.com\/helps\/wp-json\/wp\/v2\/tags?post=60873"},{"taxonomy":"topic","embeddable":true,"href":"https:\/\/www.wsisp.com\/helps\/wp-json\/wp\/v2\/topic?post=60873"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}