Pytorch实现的一版苏神的DialoGPT,一个基于GPT+NEZHA预训练模型的闲聊多轮对话模型,将Keras训练好的模型参数转成pytorch进行加载和infer。
A: '别爱我没结果'
B: '你这样会失去我的'
A: '失去了又能怎样'
生成回复:"我们在一起吧" / "我会好起来的"
A: "我今天腿都废了,你们过节,我搬砖"
B: "辛苦啊,圣诞节还去赚大钱了加油"
生成回复:"你也要加油,我要赚大钱" / "我也觉得,我要努力赚钱了"
A: "这个会不会聚划算"
B: "暂时没有哦"
A: "后期会不会有"
生成回复:"不确定哦" / "不会有哦"
A: "前排,鲁迷们都起床了吧"
B: "标题说助攻,但是看了那球,真是活生生的讽刺了"
生成回复:"我也看了那球,不过没说我是球迷啊,我是看球的,我也觉得鲁蜜挺可怜" / "我也看了"
A: "火锅我在重庆成都吃了七八顿火锅"
B: "哈哈哈哈!那我的嘴巴 可能要烂掉!"
生成回复:"哈哈哈哈哈你可爱死了" / "我的嘴唇也要烂掉"
# 依靠NEZHA处理长文本的能力,处理超多轮次对话
A: "在",
B: "在的亲有什么可以帮您吗",
A: "还送不送的不送货我退了",
B: "亲我们这边帮您更换其他快递了",
A: "31号到今天3天了",
B: "亲真是不好意思呢我这边再帮您催一下快递",
A: "今天不到退了什么鬼快递",
B: "真是对不起了亲",
A: "31号中午就到这边了就是不送",
B: "好的亲我这边帮您向邮政那边反应了",
A: "那我退货算了事不过三你说对吧",
B: "亲最近是元旦呢可能物流量大我明天帮您和EMS联系帮您确认一下具体情况您看可以吗",
A: "今天已经3天了",
B: "亲您都等了三天了呢我明天帮您和快递联系一下知道具体情况后您再决定可以吗不然您这3天都白等了呢",
A: "有什么办法快递都送给我了我还是去实体店买吧心累",
B: "实在是对不起您了那如果后续快递再给您送来的话麻烦您拒收一下就可以退回了亲",
A: "你点下退款吧",
B: "您稍等相应的工作人员会帮您处理的",
A: "快递不送有什么办法",
B: "好的亲",
A: "已经是马上到手的东西叫你等3天你自己想想电话都没一个来安慰一下我",
B: "真对不起了亲您的情况我会帮您记录的您下次再来我们店我们给您送点额外的小礼品",
A: "你自己说对不对",
B: "是的亲",
A: "我这次第二次在你这里买了感觉好才买谁知道快递不给力",
B: "下次麻烦您提醒一下我们给您换其他的快递",
A: "哦你退款了我重新下单",
B: "好的亲退款是相应的工作人员受理的麻烦您等待一下就可以了",
A: "好",
B: "麻烦您了亲"
生成回复:"没事了你看着办吧" / "没事" / "你们公司不给力还有理由吗"
模型下载地址百度网盘 密码: 7p5i