Skip to content

从头开始使用transfomer构建seq2seq(对对联)、自编码(新闻文本分类)、自回归(斗破苍穹小说续写)任务。

Notifications You must be signed in to change notification settings

taishan1994/transformer-examples

Repository files navigation

transformer-examples

从头开始使用transfomer构建seq2seq(对对联)、自编码(新闻文本分类)、自回归(斗破苍穹小说续写)任务。该项目可以了解transfomer每一部分的具体结构,以及加深对类似bert(只使用transformer的编码器)、GPT(只使用transformer的解码器)、BART(使用transformer的编码器和解码器)的理解。

数据及模型下载地址:

链接:https://pan.baidu.com/s/1M6c3iAnW_du_LH8tH2uCZA?pwd=4sdh
提取码:4sdh

依赖

pytorch >= 1.6+
transformers==4.5.0

运行及结果

seq2seq自动对联

python duilian_seq2seq.py

上联几缕兰香萦素袖
真实一窗竹韵养虚怀
下联一帘柳色醉清风
====================================================================================================
上联梅花不惧严冬雪
真实荷蕊敢迎酷夏阳
下联柳叶常吟落水风
====================================================================================================
上联腊月径寒神州续作梅花赋
真实春阳送暖赤县频吹尧舜风
下联春风风暖岁月长来柳叶香
====================================================================================================
上联黄鹤醉回眸武昌鱼戏水中鹤
真实楚天舒放眼鹦鹉嘴亲江里天
下联青花开入眼天湃龙飞天上花
====================================================================================================
上联烟雨婆娑一曲恐惊花溅泪
真实柳衣袅绕三更犹喜月拂弦
下联春风浩漫千秋漫起柳含诗
====================================================================================================
上联金骏眉金德性双修成大道
真实正山堂正禅茶一味致中和
下联金龙水秀人生一片展新图
====================================================================================================
上联醉倚青山揽就松风权作枕
真实惯游绿水欣邀梅雨喜冲波
下联闲游碧水欣将月色自成诗
====================================================================================================
上联镰运锤挥万古江山开盛世
真实龙嚷虎奋九州儿女创殊荣
下联文明气展千秋社会展新图
====================================================================================================
上联日曜锤镰砥砺朱毛传马列
真实风吹号角追随刘邓到江淮
下联风流盛世欣扬玉海展龙腾
====================================================================================================
上联陈水扁阿水扁陈水阿扁扁扁扁
真实秦木会汉木会秦木汉会会会会
下联清风浩滟风风清风浩滟古凰香
====================================================================================================
上联喜气满门庭亿民乐佳节
真实春风吹大地万物吐新茸
下联春风盈海韵万里展新图
====================================================================================================
上联创业艰辛业怠
真实说山容易上山难
下联兴民发展富民兴
====================================================================================================
上联阅成大海心方阔
真实攀得奇峰眼自高
下联不是青山意自高
====================================================================================================
上联喜气盈门祥老少
真实春风得意吉人家
下联春风入地绿花香

新闻文本分类

python classification_auto_encoder.py

文本:《堵车吴镇宇首度国语表演 激发话剧表演兴趣新浪娱乐讯 电影堵车将于10月14日全国公映吴镇宇在片中饰演一位名叫罗生门的环保主义者和大毒枭开着一辆红色的救护车闯荡在高速发展的城市森林吴镇宇表示自己能出演这种亦正亦邪的角色很过瘾直言罗生门不仅保留其神经帝的本质还挖掘了他最真挚的爱情观更透露该片是其第一部尝试国语配音的作品台词和飙车考验吴镇宇演技堵车由北京紫禁城影业天津华唐文化北京奇新世纪影视浙江世纪风采昆明博奥投资联合出品是导演阿年筹备5年之作谈及接拍此片的原因吴镇宇笑称是因为没看明白剧本所以想挑战一把。“直到堵车杀青我都不是很理解所扮演的人物,‘罗生门既有阴暗复杂的一面也有深情诗意的一面不同于以往塑造的反面角色是个很复杂的人物。”片中,“罗生门有大量类似莎士比亚经典戏剧的对白例如你知道吗距离我们的宿命越来越近了在U型建筑前面举行婚礼婚礼就是葬礼就是祭奠就是毁灭。”除了要背诵大段的对白吴镇宇还要亲身上演飙车戏已故动作导演薛春炜为其设计了大量飙车戏吴镇宇透露他和陈志朋倪大红孙海英(微博)等人的车斗场面将会是本片的一大看点首度为
真实标签娱乐
预测标签娱乐
====================================================================================================
文本三年来首只LOF指数基金今起发售本报讯三年来首只LOF指数基金——鹏华沪深300指数基金(LOF)今日起募集投资者可通过工招等商业银行以及国信安信湘财等各大券商进行认购由于是LOF基金投资者还可以通过深交所会员单位在场内认购基金鹏华沪深300指数基金在投资策略上将用基金净值不低于90%的资产投资沪深300指数投资目标追求基金净值增长率与业绩比较基准收益率之间的跟踪误差最小化力争将日平均跟踪误差控制在0.35%以内以实现对沪深300指数的有效跟踪。□本报记者 易非 深圳报道3月2日起三年来首只LOF指数基金——鹏华沪深300指数基金LOF开始募集投资者可通过工招等商业银行以及国信安信湘财等各大券商进行认购由于是LOF基金投资者还可以通过深交所会员单位在场内认购基金公告显示鹏华沪深300指数基金在投资策略上将用基金净值不低于90%的资产投资沪深300指数投资目标追求基金净值增长率与业绩比较基准收益率之间的跟踪误差最小化力争将日平均跟踪误差控制在0.35%以内以实现对沪深300指数的有效跟踪理财专家表示指数基金属于高风险高收益的投资品
真实标签财经
预测标签财经
====================================================================================================
文本黄蜂vs湖人首发科比保罗大对决 阿里扎反戈一击新浪体育讯北京时间4月18日凌晨3:30NBA季后赛洛杉矶湖人主场迎战新奥尔良黄蜂赛前双方也公布了首发阵容湖人队费舍尔科比阿泰斯特加索尔拜纳姆黄蜂队保罗贝里内利阿里扎兰德里奥卡福[新浪NBA官方微博][新浪NBA湖人新闻动态微博][新浪NBA专题][黄蜂vs湖人图文直播室](新浪体育)
真实标签体育
预测标签体育
====================================================================================================
文本:《手机大航海》―我的爱帆船载着我的梦想野心之大只有广阔七海方能承载。”  从那个年代走过来的玩家而言这句话绝不陌生曾几何时我们在方寸屏幕前扬帆起航去扮演形形色色的各式人等纵横海洋光荣的大航海时代系列已经镌刻于一代人的经历中不可磨灭而今手机大航海又来到我们面前随时随地就能感受到海洋美酒还有丝绸飘荡着沁人心肺的芳香远处是人鱼的呼唤我又再一次义无返顾地投入了无尽的海洋第一次玩航海类题材游戏是在初中的时候玩得大航海时代4那时候班里有电脑的家庭也不多所以如果一个人玩什么游戏那么所有人都会回去轮流安装一起玩也一起交流经验心得当第一次进入大航海时代4的时候随着鸽子的鸣镝声出现非常维美的画面那一刻我的心完全被大航海时代4所征服还为了更好的玩游戏专门买了一本世界地图册每次打开游戏时桌上就放着一本地图册草稿纸和计算器第二天到了学校还和同学相比谁赚得钱更多买得船更好这些是到现在玩游戏的经历当中最美好的记忆了所以当我看到有一款大航海题材类的网游登陆手机平台时我的心情是非常难以抑制的等到手机大航海公测的时候我急冲冲得注册游戏并加入到大航海的怀抱里!《
真实标签游戏
预测标签游戏
====================================================================================================
文本组图看达人如何演绎豹纹军装风导语懂得潮流的人总是知道自己最适合什么今冬流行的亮点太多军装豹纹类似民族风情想要知道搭配快来看看时尚之都的达人们是怎么穿的吧军绿色宽松款大衣不失俏皮味道高腰设计的短裙有效的提升了腰线衬托出修长的美腿豹纹是今年冬季最抢眼的搭配元素加上驼色针织衫灰色围巾及暖棕色手挎包整体色调统一又迷人棕色与蓝色的结合能让人眼前一亮简洁的款式依然能突显你的独特品味宽松的针织外套衬托出优美的身形即使搭配同样沉闷的黑色包包性感的丝袜装扮都依然能透露出迷人的气息小立领衬衫加上深黄高腰裤摩登感十足随意披上的外套更显慵懒又个性的法式风情真实标签时尚
预测标签时尚
====================================================================================================
文本组图穿日本内衣扮演他的小辣椒导语日本的少女内衣设计一向令人赞叹不已女孩子注重的甜美可爱与成熟华丽感并存使内衣颇具艺术感不但颜色丰富质料也是布质丝质等十分丰富更会配合流行色和流行款式推出很多难以想象的设计Ravijour内衣品牌虽然价格上比起一般内衣要贵但设计非常性感能让你的芳心一下就被虏获真实标签时尚
预测标签时尚
====================================================================================================
文本公寓不达标 在澳留学生被勒令搬离难获紧急帮助中新网(微博)11月3日电 据澳大利亚新快网报道有机构报称出于安全隐患一批阿德莱德市的在澳留学生被勒令搬离住所但当中仅有27人获得帮助居住在王子公寓(Princes Apartments)及Regency公寓的人大部分是留学生他们被告知要在周末前搬走却可能不能获得紧急住宿帮助阿德莱德市地方政府的发言人表示地方政府已经通知业主这栋物业不符合标准。“这栋物业的前业主已经破产所以政府于今年10月把这些违规告知接管人Ferrier Hodgson。”阿德莱德教育部门的总裁Denise von Wald表示会严肃对待这种情况该部门正进行调查为学生寻找新的住处她说:“到目前为止据我们所知南澳有三所公立大学卡内基美隆大学(Carnegie Mellon University)和南澳技术及继续教育部门一共收到了27名学生的申请他们报告了在王子公寓及Regent公寓的住宿问题要求我们协助他们找到新的住处确实据我了解这是阿德莱德第一次出现这类事件同时这也是人们不可接受的事情。”Wald表示培训支持办事处(Office of the
真实标签教育
预测标签教育
====================================================================================================
文本詹姆斯盼率队完美收官 皇帝我们可击败任何对手新浪体育讯北京时间4月5日(迈阿密时间4月4日)消息  在常规赛还剩下5场比赛的时候热火的目标已经逐渐清晰那就是继续赢球坐稳东部第二以确保季后赛半决赛的主场优势。“我们喜欢打客场比赛在客场不管是场上还是场下我们球队都紧紧团结在一块,”勒布朗-詹姆斯在热火以三连胜结束四连客征程后如是感慨说他们本赛季的客场战绩为26胜13负高居东部第一在全联盟暂时仅次于27胜13负的小牛,“当然我们也需要把这种感觉带回主场。”詹姆斯之所以这么说与他们的主场战绩不那么出众不无关系28胜10负别说比东部的竞争对手绿军(30胜8负)和公牛(33胜5负)甚至都比不过掘金(31胜7负)。“我们在主场还可变得更好,”迈克-米勒说,“我们在客场打得更好了但在季后赛我们必须保护主场。”结束四连客征程之后热火回到了迈阿密他们将先打三个主场对手分别是雄鹿山猫和绿军主场与绿军一战很可能就是事关东部争二的生死战。“我们都很期待下周日与波士顿下周一与亚特兰大的比赛,”德维恩-韦德说,“当这个赛季结束的时候你就会发现与某些球队的比赛总是不好打而我们现在需要做的就
真实标签体育
预测标签体育
====================================================================================================
文本时代数码广场理光R10数码相机特价优惠 理光R10是继理光R8的成功推出后的后继机型除了像素等相应的提高之外也增加了一些新颖的功能颇具特色的理光操作风格也在R10身上得到很多的改进其古典的造型和大变焦比的镜头都是其主要卖点笔者日前在市场了解到该款相机的报价为2650元喜欢的朋友可以关注一下外观上理光R10依然采用的复古的外观设计顶盖的设计也和R8一样采用金属材质制成显得美观坚固而静心设计的手柄也让消费者更容易握持而且手柄的设计简洁美观实用;在配色上理光R10除了有复古的银色和神秘的黑色外, 还有充满个性的啡金色所以能极大的满足消费者的的喜好机身配置上理光R10配备了一块1/2.3英寸1000万有效像素的CCD感光元件并与适合各种拍摄条件的图像处理引擎——平滑图像引擎III融为一体而且理光R10最为成功的便是将7.1倍光学变焦镜头装配在了最薄点仅为22.6 mm的机身上焦距等效35mm相机的28-200mm并可按七级固定焦距更改光学变焦放大倍数最近的对焦距离为1cm机身背面配备的是一块3.0英寸46万像素的HVGA显示屏拥有广阔的视角和高对比度
真实标签科技
预测标签科技
====================================================================================================
文本组图明星示范秋冬必败十大流行款导语2010年秋冬季时尚圈什么单品正被热推呢蕾丝衬衣皮草马甲牛仔外套驼色大衣……似乎款款都能具有让你变得更加时髦的理由明星们早已经抢先我们一步尝试了今年秋冬最新鲜的造型就让她们来告诉你旧时光中的优雅女性终于重现在人们面前刻意提高的紧身腰线以及优美如波浪般向两侧延展的蓬松伞裙都是为了突出腰肢的丰满曲线并且可以完美掩盖下肢的缺陷如果你小腿还算匀称何不尝试一下简约的白衬衫打造出Effortless Chic跟着极简主义的步调挽起袖子立起领子再搭上一件九分裤即刻变身街拍中性女王一件保暖的针织毛衣能让你度过一个温暖幸福的秋季搭着复古的风潮衣橱那件深藏已久的毛衣带着温暖质感宽松随意却又不失亲和力入时洒脱的牛仔街头永远时髦的单品一件牛仔外套无论搭配上什么单品都会让你看起来年轻不少帅气的连身裤最近大热好莱坞看来这股风潮铁定要延续到秋冬显然长裤在秋冬更受宠如果你是走优雅路线的那么可以尝试下Jessica Beil这件无袖式Jumpsuit如果你是走朋克路线的则可以尝试一下Chole的黑色Jumpsuit真实标签时尚
预测标签时尚
====================================================================================================

              precision    recall  f1-score   support

          体育       0.98      0.97      0.97      1000
          财经       0.95      0.92      0.93      1000
          房产       0.70      0.83      0.76      1000
          家居       0.88      0.53      0.66      1000
          教育       0.89      0.85      0.87      1000
          科技       0.84      0.96      0.89      1000
          时尚       0.92      0.92      0.92      1000
          时政       0.83      0.90      0.86      1000
          游戏       0.93      0.88      0.90      1000
          娱乐       0.88      0.97      0.92      1000

    accuracy                           0.87     10000
   macro avg       0.88      0.87      0.87     10000
weighted avg       0.88      0.87      0.87     10000

斗破苍穹小说续写

python gnerate_auo_regression.py

开头:“咳咳眼瞳猛然睁开萧炎急忙低下头
真实:“咳咳眼瞳猛然睁开萧炎急忙低下头剧烈的咳嗽着半晌后方才脸色涨红的抬起来头来望着身后那正端着一碗清水表情有些不知所措的云芝嘴角微微抽搐苦笑道:“你成心把我呛死是吧生成:“咳咳眼瞳猛然睁开萧炎急忙低下头来看手掌一把抓住了一个小小的玉瓶顿时一股股极为恐怖的劲力将之渲染得犹如一个玉般将之上的一团团团团团团团团团团团团团团团团团团团团团尽数包裹而进<eos>
====================================================================================================

开头凶悍能量在虚空狠狠撞击爆发出一道道嗤嗤
真实凶悍能量在虚空狠狠撞击爆发出一道道嗤嗤声响一阵淡淡的烟雾也是升腾而起生成凶悍能量在虚空狠狠撞击爆发出一道道嗤嗤的声响在天际响彻而起那股无形的能量也是在此刻犹如受到一种极为恐怖的威压不断的从天空上涌出这种威压令得人头皮发麻麻的脸庞都是忍不住的有些变得有些变色了起来<eos>
====================================================================================================

开头然而就在所有人都以为萧炎定然难逃一劫之时
真实然而就在所有人都以为萧炎定然难逃一劫之时天空之上一股磅礴劲气涟漪猛然以萧炎为中心点暴涌而出只见劲气所过之处两道即使是连海波东都难以正面抵挡的分身残影却是骤然凝固旋即在下方一道道呆滞的目光中一声闷响轰然爆裂
生成然而就在所有人都以为萧炎定然难逃一劫之时那一霎一道道惊骇目光却是突然从那道惊天动地的目光中闪过一道惊骇之色旋即旋即一道道黑色身影缓缓浮现而出<eos>
====================================================================================================

开头:“真实:“生成:“<eos>
====================================================================================================

开头弯月刃带起一道道刺耳的音爆之声划破天际
真实弯月刃带起一道道刺耳的音爆之声划破天际那股一往无前的强悍威势甚至是有种要将天空横劈为两半的势头生成弯月刃带起一道道刺耳的音爆之声划破天际最后在那一道道目光注视下缓缓旋转最后停留在了那片区域的地方那里一道道目光望着远处的一座山峰上那里一片空间都是出现了一道道目光显得格外的人在这般平静的气氛下显然这些人心中都是有着一种极为不错的感觉<eos>
====================================================================================================

开头花了将近十分钟萧炎方才走完这一条走廊真实花了将近十分钟萧炎方才走完这一条走廊虽然其中并没有他所需要的龙须冰火果但是其他那一种种极为稀罕的药材却是让得萧炎心脏的跳动加速了许多生成花了将近十分钟萧炎方才走完这一条走廊然后在一旁的石台上望着那些石台一眼望去一些人都是忍不住的跳了跳这种情况即便是他们都是未曾见到但这些人却是一个人正是那些家伙的弟子<eos>
====================================================================================================

开头:“你便是萧炎炎盟的那个盟主狮天目光死死
真实:“你便是萧炎炎盟的那个盟主狮天目光死死的盯着萧炎先前后者抱彩鳞时也是被其看见当时心中妒火便是猛烈的升腾了起来即便现在尚还未攻破玄黄要塞但他已经是将彩鳞当成了他的禁脔怎容他人染指
生成:“你便是萧炎炎盟的那个盟主狮天目光死死的盯着那道黑色光柱萧炎眼中也是涌现一抹惊讶这个家伙实在是太过恐怖了点<eos>
====================================================================================================

开头被同伴拉扯了一下那中年大汉也是从游身尺
真实被同伴拉扯了一下那中年大汉也是从游身尺所带来的狂热中清醒了不少目光望向萧炎那边刚好是瞧得小医仙那淡漠视线当下浑身便是一个哆嗦咽了口唾沫只得悻悻的坐了回去生成被同伴拉扯了一下那中年大汉也是从游身尺上跃下目光在萧炎身上扫过最后停留在了那张熟悉的脸庞上目光扫向那张略有些熟悉的脸庞上此刻的他正是那张年轻脸庞此刻的少年此刻正是在这种年轻人他们的感觉到似乎比起当年的那种恐怖还要更加的让得人有些难以忍受<eos>
====================================================================================================

开头就在萧炎皱眉间突然间一道极其细微的呻
真实就在萧炎皱眉间突然间一道极其细微的呻吟之声透过水流传进了其耳中生成就在萧炎皱眉间突然间一道极其细微的呻吟声突然在其心中响起旋即一道道惊骇目光猛然转向了天空上的那道身影正是那位黑袍青年<eos>
====================================================================================================

开头:“云岚宗对我萧家所做之事我会百倍尝还真实:“云岚宗对我萧家所做之事我会百倍尝还今曰便先由你二人开始吧放心云山老狗也会下去陪你们的。”
生成:“云岚宗对我萧家所做之事我会百倍尝还你们这些年也是在这里你们也不是省油的灯而且还有着一些实力强横的势力我们也不敢与你们抗衡所以我们也不多管闲你们一起我们便是要将你们的身份我们之间的事便是我们不会让得你们有些麻烦。”萧炎淡淡的道<eos>
====================================================================================================

效果不是很好,毕竟构建的模型不是很大。

参考

A PyTorch implementation of the Transformer model in "Attention is All You Need"

About

从头开始使用transfomer构建seq2seq(对对联)、自编码(新闻文本分类)、自回归(斗破苍穹小说续写)任务。

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages