Transformer-based models implemented in tensorflow 2.x(Keras)
Project description
transformers-keras
基于tf.keras
的Transformers系列模型实现。
所有的Model
都是keras模型,可以直接用于训练模型、评估模型或者导出模型用于部署。
在线文档:transformers-keras文档
本库功能预览:
- 加载各种预训练模型的权重
- 掩码语言模型(Masked Language Model)解决方案
- 句向量(Sentence Embedding)的解决方案
- 抽取式问答任务(Question Answering)的解决方案
- 序列分类任务(Sequence Classification)的解决方案
- 序列标注任务(Token Classification)的解决方案
- 知识蒸馏(Knowledge Distillation)的框架
以上任务的数据处理、模型构建都完成了,可以一键开始训练。这里有两个例子:
更多使用方法和介绍,请查看在线文档:transformers-keras文档
联系方式
由于二维码过期限制,加微信讨论群请先加我微信邀请加入。
- 微信: luozhouyang0528
- 邮箱: zhouyang.luo@gmail.com
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
transformers_keras-0.5.0.tar.gz
(39.1 kB
view hashes)
Built Distribution
Close
Hashes for transformers_keras-0.5.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 252e48c08f8ed4b13da1c9c6f2216e3d10d4938ed1a4681765692637a31b9ffe |
|
MD5 | 39f0c0bc21d3e6ad9e5ca24950ccefae |
|
BLAKE2b-256 | 3d4abcb3a8b807ac1aa0b004ef822ee43d6b77ea975e42977d5c36e3e63fbc1a |