Skip to main content

Transformers kit - Multi-task QA/Tagging/Multi-label Multi-Class Classification/Generation with BERT/ALBERT/T5/BERT

Project description




PyPI Download Build Last Commit CodeFactor Visitor

What is it

TFKit is a deep natural language process framework for classification/tagging/question answering/embedding study and language generation.
It leverages the use of transformers on many tasks with different models in this all-in-one framework.
All you need is a little change of config.

Task Supported

With transformer models - BERT/ALBERT/T5/BART......

Classification multi-class and multi-label classification
Question Answering extractive qa
Question Answering multiple-choice qa
Tagging sequence level tagging / sequence level with crf
Text Generation seq2seq language model
Text Generation causal language model
Text Generation once generation model / once generation model with ctc loss
Text Generation onebyone generation model
Self-supervise Learning mask language model

Getting Started

Learn more from the document.

Supplement

Contributing

Thanks for your interest.There are many ways to contribute to this project. Get started here.

License PyPI - License

Icons reference

Icons modify from Freepik from www.flaticon.com
Icons modify from Nikita Golubev from www.flaticon.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

tfkit-0.7.18.tar.gz (214.6 kB view hashes)

Uploaded Source

Built Distributions

tfkit-0.7.18-py3.7.egg (175.1 kB view hashes)

Uploaded Source

tfkit-0.7.18-py3-none-any.whl (78.0 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