Skip to main content

Transformer-based models implemented in tensorflow 2.x(Keras)

Project description

transformers-keras

Python package PyPI version Python

基于tf.keras的Transformers系列模型实现。

所有的Model都是keras模型,可以直接用于训练模型、评估模型或者导出模型用于部署。

在线文档:transformers-keras文档

本库功能预览:

  • 加载各种预训练模型的权重
  • 掩码语言模型(Masked Language Model)解决方案
  • 句向量(Sentence Embedding)的解决方案
  • 抽取式问答任务(Question Answering)的解决方案
  • 序列分类任务(Sequence Classification)的解决方案
  • 序列标注任务(Token Classification)的解决方案
  • 知识蒸馏(Knowledge Distillation)的框架

以上任务的数据处理、模型构建都完成了,可以一键开始训练。这里有两个例子:

更多使用方法和介绍,请查看在线文档:transformers-keras文档

联系方式

由于二维码过期限制,加微信讨论群请先加我微信邀请加入。

Project details


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.1.tar.gz (39.1 kB view hashes)

Uploaded Source

Built Distribution

transformers_keras-0.5.1-py3-none-any.whl (63.1 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page