├── .github ├── ISSUE_TEMPLATE │ ├── bug_report.md │ ├── feature_request.md │ └── question.md └── workflows │ └── ci.yml ├── .gitignore ├── .pre-commit-config.yaml ├── AUTHORS.rst ├── CHANGELOG.rst ├── LICENSE.txt ├── MANIFEST.in ├── Makefile ├── README.md ├── docs ├── .templates │ ├── config.mako │ └── html.mako ├── biome_text_logo_for_readme.png ├── docs │ ├── .vuepress │ │ ├── config.js │ │ ├── public │ │ │ ├── assets │ │ │ │ ├── fonts │ │ │ │ │ ├── BasisGrotesquePro-Bold.woff │ │ │ │ │ ├── BasisGrotesquePro-Light.woff │ │ │ │ │ ├── BasisGrotesquePro-Regular.woff │ │ │ │ │ └── justmeagaindownhere.woff │ │ │ │ └── img │ │ │ │ │ ├── allennlp.png │ │ │ │ │ ├── allennlp.svg │ │ │ │ │ ├── bg.svg │ │ │ │ │ ├── biome-isotype.svg │ │ │ │ │ ├── biome.svg │ │ │ │ │ ├── hugging.png │ │ │ │ │ ├── huggingface.svg │ │ │ │ │ ├── pytorch.svg │ │ │ │ │ └── recognai.png │ │ │ └── favicon.ico │ │ └── theme │ │ │ ├── components │ │ │ ├── AlgoliaSearchBox.vue │ │ │ ├── Home.vue │ │ │ ├── NavLink.vue │ │ │ ├── Navbar.vue │ │ │ ├── PageNav.vue │ │ │ ├── Sidebar.vue │ │ │ ├── Versions.vue │ │ │ ├── search-orange.svg │ │ │ └── search.svg │ │ │ ├── enhanceApp.js │ │ │ ├── index.js │ │ │ ├── layouts │ │ │ ├── 404.vue │ │ │ └── Layout.vue │ │ │ └── styles │ │ │ ├── code-colors.styl │ │ │ ├── fonts.styl │ │ │ ├── index.styl │ │ │ └── palette.styl │ ├── README.md │ ├── api │ │ └── README.md │ ├── documentation │ │ ├── basics.md │ │ ├── community │ │ │ ├── 1-contributing.md │ │ │ ├── 2-get_help.md │ │ │ └── 3-developer_guides.md │ │ ├── readme.md │ │ ├── tutorials │ │ │ ├── 1-Training_a_text_classifier.ipynb │ │ │ ├── 2-Training_a_sequence_tagger_for_Slot_Filling.ipynb │ │ │ ├── 3-Hyperparameter_optimization_with_Ray_Tune.ipynb │ │ │ ├── 4-Using_Transformers_in_biome_text.ipynb │ │ │ └── img │ │ │ │ ├── analysis_df.png │ │ │ │ ├── hpo_tensorboard.png │ │ │ │ └── text_classifier_explore_screenshot.png │ │ └── user-guides │ │ │ ├── 1-nlp-tasks.md │ │ │ ├── 2-configuration.md │ │ │ └── 3-deployment.md │ └── icons │ │ ├── blank.svg │ │ ├── chev-left.svg │ │ └── chev-right.svg ├── internal_docs │ └── configs.md ├── package.json └── prepare_versioned_build.sh ├── environment_dev.yml ├── setup.cfg ├── setup.py ├── src └── biome │ ├── __init__.py │ └── text │ ├── __init__.py │ ├── backbone.py │ ├── cli │ ├── __init__.py │ ├── cli.py │ ├── evaluate.py │ ├── serve.py │ └── train.py │ ├── commons.py │ ├── configuration.py │ ├── dataset.py │ ├── errors.py │ ├── features.py │ ├── featurizer.py │ ├── helpers.py │ ├── hpo.py │ ├── metrics.py │ ├── mlflow_model.py │ ├── model.py │ ├── modules │ ├── __init__.py │ ├── configuration │ │ ├── __init__.py │ │ ├── allennlp_configuration.py │ │ └── defs.py │ ├── encoders │ │ ├── __init__.py │ │ └── time_distributed_encoder.py │ └── heads │ │ ├── __init__.py │ │ ├── classification │ │ ├── __init__.py │ │ ├── classification.py │ │ ├── doc_classification.py │ │ ├── record_classification.py │ │ ├── record_pair_classification.py │ │ ├── relation_classification.py │ │ └── text_classification.py │ │ ├── language_modelling.py │ │ ├── task_head.py │ │ ├── task_prediction.py │ │ └── token_classification.py │ ├── pipeline.py │ ├── text_cleaning.py │ ├── tokenizer.py │ ├── trainer.py │ └── vocabulary.py └── tests ├── __init__.py ├── conftest.py ├── docs ├── __init__.py ├── test_configurations.py └── test_tutorials.py ├── resources └── data │ ├── business.cat.2k.train.csv │ ├── business.cat.2k.valid.csv │ ├── dataset_sequence.json │ ├── dataset_sequence.jsonl │ ├── dataset_source.csv │ ├── dataset_source.jsonl │ ├── emotions_with_transformers.txt │ ├── nested-list.jsonl │ ├── test.parquet │ ├── test.xlsx │ └── to-be-flattened.jsonl ├── text ├── __init__.py ├── modules │ ├── __init__.py │ ├── configuration │ │ └── test_component_configuration.py │ └── heads │ │ ├── __init__.py │ │ ├── classification │ │ ├── __init__.py │ │ ├── test_document_classification.py │ │ ├── test_record_pair_classification.py │ │ ├── test_relation_classifier.py │ │ └── test_text_classification.py │ │ ├── test_language_modelling.py │ │ ├── test_task_head.py │ │ └── test_token_classification.py ├── test_cli.py ├── test_commons.py ├── test_configuration.py ├── test_dataset.py ├── test_features_configuration.py ├── test_features_transformers.py ├── test_hpo.py ├── test_metrics.py ├── test_model_predict.py ├── test_pipeline_copy.py ├── test_pipeline_datasets.py ├── test_pipeline_predict.py ├── test_pipeline_save.py ├── test_pipeline_to_mlflow.py ├── test_pipeline_tokenizer.py ├── test_pipeline_vocab.py ├── test_pipeline_with_custom_head.py ├── test_pipeline_with_optional_inputs.py ├── test_pretrained_word_vectors.py ├── test_text_cleaning.py ├── test_tokenizer.py └── test_trainer.py └── text_classification_integration_test.py /.github/ISSUE_TEMPLATE/bug_report.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/.github/ISSUE_TEMPLATE/bug_report.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature_request.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/.github/ISSUE_TEMPLATE/feature_request.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/question.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/.github/ISSUE_TEMPLATE/question.md -------------------------------------------------------------------------------- /.github/workflows/ci.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/.github/workflows/ci.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /AUTHORS.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/AUTHORS.rst -------------------------------------------------------------------------------- /CHANGELOG.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/CHANGELOG.rst -------------------------------------------------------------------------------- /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/README.md -------------------------------------------------------------------------------- /docs/.templates/config.mako: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/.templates/config.mako -------------------------------------------------------------------------------- /docs/.templates/html.mako: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/.templates/html.mako -------------------------------------------------------------------------------- /docs/biome_text_logo_for_readme.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/biome_text_logo_for_readme.png -------------------------------------------------------------------------------- /docs/docs/.vuepress/config.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/.vuepress/config.js -------------------------------------------------------------------------------- /docs/docs/.vuepress/public/assets/fonts/BasisGrotesquePro-Bold.woff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/.vuepress/public/assets/fonts/BasisGrotesquePro-Bold.woff -------------------------------------------------------------------------------- /docs/docs/.vuepress/public/assets/fonts/BasisGrotesquePro-Light.woff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/.vuepress/public/assets/fonts/BasisGrotesquePro-Light.woff -------------------------------------------------------------------------------- /docs/docs/.vuepress/public/assets/fonts/BasisGrotesquePro-Regular.woff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/.vuepress/public/assets/fonts/BasisGrotesquePro-Regular.woff -------------------------------------------------------------------------------- /docs/docs/.vuepress/public/assets/fonts/justmeagaindownhere.woff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/.vuepress/public/assets/fonts/justmeagaindownhere.woff -------------------------------------------------------------------------------- /docs/docs/.vuepress/public/assets/img/allennlp.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/.vuepress/public/assets/img/allennlp.png -------------------------------------------------------------------------------- /docs/docs/.vuepress/public/assets/img/allennlp.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/.vuepress/public/assets/img/allennlp.svg -------------------------------------------------------------------------------- /docs/docs/.vuepress/public/assets/img/bg.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/.vuepress/public/assets/img/bg.svg -------------------------------------------------------------------------------- /docs/docs/.vuepress/public/assets/img/biome-isotype.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/.vuepress/public/assets/img/biome-isotype.svg -------------------------------------------------------------------------------- /docs/docs/.vuepress/public/assets/img/biome.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/.vuepress/public/assets/img/biome.svg -------------------------------------------------------------------------------- /docs/docs/.vuepress/public/assets/img/hugging.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/.vuepress/public/assets/img/hugging.png -------------------------------------------------------------------------------- /docs/docs/.vuepress/public/assets/img/huggingface.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/.vuepress/public/assets/img/huggingface.svg -------------------------------------------------------------------------------- /docs/docs/.vuepress/public/assets/img/pytorch.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/.vuepress/public/assets/img/pytorch.svg -------------------------------------------------------------------------------- /docs/docs/.vuepress/public/assets/img/recognai.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/.vuepress/public/assets/img/recognai.png -------------------------------------------------------------------------------- /docs/docs/.vuepress/public/favicon.ico: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/.vuepress/public/favicon.ico -------------------------------------------------------------------------------- /docs/docs/.vuepress/theme/components/AlgoliaSearchBox.vue: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/.vuepress/theme/components/AlgoliaSearchBox.vue -------------------------------------------------------------------------------- /docs/docs/.vuepress/theme/components/Home.vue: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/.vuepress/theme/components/Home.vue -------------------------------------------------------------------------------- /docs/docs/.vuepress/theme/components/NavLink.vue: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/.vuepress/theme/components/NavLink.vue -------------------------------------------------------------------------------- /docs/docs/.vuepress/theme/components/Navbar.vue: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/.vuepress/theme/components/Navbar.vue -------------------------------------------------------------------------------- /docs/docs/.vuepress/theme/components/PageNav.vue: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/.vuepress/theme/components/PageNav.vue -------------------------------------------------------------------------------- /docs/docs/.vuepress/theme/components/Sidebar.vue: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/.vuepress/theme/components/Sidebar.vue -------------------------------------------------------------------------------- /docs/docs/.vuepress/theme/components/Versions.vue: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/.vuepress/theme/components/Versions.vue -------------------------------------------------------------------------------- /docs/docs/.vuepress/theme/components/search-orange.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/.vuepress/theme/components/search-orange.svg -------------------------------------------------------------------------------- /docs/docs/.vuepress/theme/components/search.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/.vuepress/theme/components/search.svg -------------------------------------------------------------------------------- /docs/docs/.vuepress/theme/enhanceApp.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/.vuepress/theme/enhanceApp.js -------------------------------------------------------------------------------- /docs/docs/.vuepress/theme/index.js: -------------------------------------------------------------------------------- 1 | module.exports = { 2 | extend: '@vuepress/theme-default', 3 | } 4 | -------------------------------------------------------------------------------- /docs/docs/.vuepress/theme/layouts/404.vue: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/.vuepress/theme/layouts/404.vue -------------------------------------------------------------------------------- /docs/docs/.vuepress/theme/layouts/Layout.vue: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/.vuepress/theme/layouts/Layout.vue -------------------------------------------------------------------------------- /docs/docs/.vuepress/theme/styles/code-colors.styl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/.vuepress/theme/styles/code-colors.styl -------------------------------------------------------------------------------- /docs/docs/.vuepress/theme/styles/fonts.styl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/.vuepress/theme/styles/fonts.styl -------------------------------------------------------------------------------- /docs/docs/.vuepress/theme/styles/index.styl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/.vuepress/theme/styles/index.styl -------------------------------------------------------------------------------- /docs/docs/.vuepress/theme/styles/palette.styl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/.vuepress/theme/styles/palette.styl -------------------------------------------------------------------------------- /docs/docs/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/README.md -------------------------------------------------------------------------------- /docs/docs/api/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/api/README.md -------------------------------------------------------------------------------- /docs/docs/documentation/basics.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/documentation/basics.md -------------------------------------------------------------------------------- /docs/docs/documentation/community/1-contributing.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/documentation/community/1-contributing.md -------------------------------------------------------------------------------- /docs/docs/documentation/community/2-get_help.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/documentation/community/2-get_help.md -------------------------------------------------------------------------------- /docs/docs/documentation/community/3-developer_guides.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/documentation/community/3-developer_guides.md -------------------------------------------------------------------------------- /docs/docs/documentation/readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/documentation/readme.md -------------------------------------------------------------------------------- /docs/docs/documentation/tutorials/1-Training_a_text_classifier.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/documentation/tutorials/1-Training_a_text_classifier.ipynb -------------------------------------------------------------------------------- /docs/docs/documentation/tutorials/2-Training_a_sequence_tagger_for_Slot_Filling.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/documentation/tutorials/2-Training_a_sequence_tagger_for_Slot_Filling.ipynb -------------------------------------------------------------------------------- /docs/docs/documentation/tutorials/3-Hyperparameter_optimization_with_Ray_Tune.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/documentation/tutorials/3-Hyperparameter_optimization_with_Ray_Tune.ipynb -------------------------------------------------------------------------------- /docs/docs/documentation/tutorials/4-Using_Transformers_in_biome_text.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/documentation/tutorials/4-Using_Transformers_in_biome_text.ipynb -------------------------------------------------------------------------------- /docs/docs/documentation/tutorials/img/analysis_df.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/documentation/tutorials/img/analysis_df.png -------------------------------------------------------------------------------- /docs/docs/documentation/tutorials/img/hpo_tensorboard.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/documentation/tutorials/img/hpo_tensorboard.png -------------------------------------------------------------------------------- /docs/docs/documentation/tutorials/img/text_classifier_explore_screenshot.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/documentation/tutorials/img/text_classifier_explore_screenshot.png -------------------------------------------------------------------------------- /docs/docs/documentation/user-guides/1-nlp-tasks.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/documentation/user-guides/1-nlp-tasks.md -------------------------------------------------------------------------------- /docs/docs/documentation/user-guides/2-configuration.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/documentation/user-guides/2-configuration.md -------------------------------------------------------------------------------- /docs/docs/documentation/user-guides/3-deployment.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/documentation/user-guides/3-deployment.md -------------------------------------------------------------------------------- /docs/docs/icons/blank.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/icons/blank.svg -------------------------------------------------------------------------------- /docs/docs/icons/chev-left.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/icons/chev-left.svg -------------------------------------------------------------------------------- /docs/docs/icons/chev-right.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/docs/icons/chev-right.svg -------------------------------------------------------------------------------- /docs/internal_docs/configs.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/internal_docs/configs.md -------------------------------------------------------------------------------- /docs/package.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/package.json -------------------------------------------------------------------------------- /docs/prepare_versioned_build.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/docs/prepare_versioned_build.sh -------------------------------------------------------------------------------- /environment_dev.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/environment_dev.yml -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/setup.py -------------------------------------------------------------------------------- /src/biome/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/__init__.py -------------------------------------------------------------------------------- /src/biome/text/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/__init__.py -------------------------------------------------------------------------------- /src/biome/text/backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/backbone.py -------------------------------------------------------------------------------- /src/biome/text/cli/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/cli/__init__.py -------------------------------------------------------------------------------- /src/biome/text/cli/cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/cli/cli.py -------------------------------------------------------------------------------- /src/biome/text/cli/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/cli/evaluate.py -------------------------------------------------------------------------------- /src/biome/text/cli/serve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/cli/serve.py -------------------------------------------------------------------------------- /src/biome/text/cli/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/cli/train.py -------------------------------------------------------------------------------- /src/biome/text/commons.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/commons.py -------------------------------------------------------------------------------- /src/biome/text/configuration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/configuration.py -------------------------------------------------------------------------------- /src/biome/text/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/dataset.py -------------------------------------------------------------------------------- /src/biome/text/errors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/errors.py -------------------------------------------------------------------------------- /src/biome/text/features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/features.py -------------------------------------------------------------------------------- /src/biome/text/featurizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/featurizer.py -------------------------------------------------------------------------------- /src/biome/text/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/helpers.py -------------------------------------------------------------------------------- /src/biome/text/hpo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/hpo.py -------------------------------------------------------------------------------- /src/biome/text/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/metrics.py -------------------------------------------------------------------------------- /src/biome/text/mlflow_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/mlflow_model.py -------------------------------------------------------------------------------- /src/biome/text/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/model.py -------------------------------------------------------------------------------- /src/biome/text/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/biome/text/modules/configuration/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/modules/configuration/__init__.py -------------------------------------------------------------------------------- /src/biome/text/modules/configuration/allennlp_configuration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/modules/configuration/allennlp_configuration.py -------------------------------------------------------------------------------- /src/biome/text/modules/configuration/defs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/modules/configuration/defs.py -------------------------------------------------------------------------------- /src/biome/text/modules/encoders/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/modules/encoders/__init__.py -------------------------------------------------------------------------------- /src/biome/text/modules/encoders/time_distributed_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/modules/encoders/time_distributed_encoder.py -------------------------------------------------------------------------------- /src/biome/text/modules/heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/modules/heads/__init__.py -------------------------------------------------------------------------------- /src/biome/text/modules/heads/classification/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/biome/text/modules/heads/classification/classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/modules/heads/classification/classification.py -------------------------------------------------------------------------------- /src/biome/text/modules/heads/classification/doc_classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/modules/heads/classification/doc_classification.py -------------------------------------------------------------------------------- /src/biome/text/modules/heads/classification/record_classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/modules/heads/classification/record_classification.py -------------------------------------------------------------------------------- /src/biome/text/modules/heads/classification/record_pair_classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/modules/heads/classification/record_pair_classification.py -------------------------------------------------------------------------------- /src/biome/text/modules/heads/classification/relation_classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/modules/heads/classification/relation_classification.py -------------------------------------------------------------------------------- /src/biome/text/modules/heads/classification/text_classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/modules/heads/classification/text_classification.py -------------------------------------------------------------------------------- /src/biome/text/modules/heads/language_modelling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/modules/heads/language_modelling.py -------------------------------------------------------------------------------- /src/biome/text/modules/heads/task_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/modules/heads/task_head.py -------------------------------------------------------------------------------- /src/biome/text/modules/heads/task_prediction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/modules/heads/task_prediction.py -------------------------------------------------------------------------------- /src/biome/text/modules/heads/token_classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/modules/heads/token_classification.py -------------------------------------------------------------------------------- /src/biome/text/pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/pipeline.py -------------------------------------------------------------------------------- /src/biome/text/text_cleaning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/text_cleaning.py -------------------------------------------------------------------------------- /src/biome/text/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/tokenizer.py -------------------------------------------------------------------------------- /src/biome/text/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/trainer.py -------------------------------------------------------------------------------- /src/biome/text/vocabulary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/src/biome/text/vocabulary.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/__init__.py -------------------------------------------------------------------------------- /tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/conftest.py -------------------------------------------------------------------------------- /tests/docs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/docs/test_configurations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/docs/test_configurations.py -------------------------------------------------------------------------------- /tests/docs/test_tutorials.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/docs/test_tutorials.py -------------------------------------------------------------------------------- /tests/resources/data/business.cat.2k.train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/resources/data/business.cat.2k.train.csv -------------------------------------------------------------------------------- /tests/resources/data/business.cat.2k.valid.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/resources/data/business.cat.2k.valid.csv -------------------------------------------------------------------------------- /tests/resources/data/dataset_sequence.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/resources/data/dataset_sequence.json -------------------------------------------------------------------------------- /tests/resources/data/dataset_sequence.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/resources/data/dataset_sequence.jsonl -------------------------------------------------------------------------------- /tests/resources/data/dataset_source.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/resources/data/dataset_source.csv -------------------------------------------------------------------------------- /tests/resources/data/dataset_source.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/resources/data/dataset_source.jsonl -------------------------------------------------------------------------------- /tests/resources/data/emotions_with_transformers.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/resources/data/emotions_with_transformers.txt -------------------------------------------------------------------------------- /tests/resources/data/nested-list.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/resources/data/nested-list.jsonl -------------------------------------------------------------------------------- /tests/resources/data/test.parquet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/resources/data/test.parquet -------------------------------------------------------------------------------- /tests/resources/data/test.xlsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/resources/data/test.xlsx -------------------------------------------------------------------------------- /tests/resources/data/to-be-flattened.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/resources/data/to-be-flattened.jsonl -------------------------------------------------------------------------------- /tests/text/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/text/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/text/modules/configuration/test_component_configuration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/text/modules/configuration/test_component_configuration.py -------------------------------------------------------------------------------- /tests/text/modules/heads/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/text/modules/heads/classification/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/text/modules/heads/classification/test_document_classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/text/modules/heads/classification/test_document_classification.py -------------------------------------------------------------------------------- /tests/text/modules/heads/classification/test_record_pair_classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/text/modules/heads/classification/test_record_pair_classification.py -------------------------------------------------------------------------------- /tests/text/modules/heads/classification/test_relation_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/text/modules/heads/classification/test_relation_classifier.py -------------------------------------------------------------------------------- /tests/text/modules/heads/classification/test_text_classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/text/modules/heads/classification/test_text_classification.py -------------------------------------------------------------------------------- /tests/text/modules/heads/test_language_modelling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/text/modules/heads/test_language_modelling.py -------------------------------------------------------------------------------- /tests/text/modules/heads/test_task_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/text/modules/heads/test_task_head.py -------------------------------------------------------------------------------- /tests/text/modules/heads/test_token_classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/text/modules/heads/test_token_classification.py -------------------------------------------------------------------------------- /tests/text/test_cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/text/test_cli.py -------------------------------------------------------------------------------- /tests/text/test_commons.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/text/test_commons.py -------------------------------------------------------------------------------- /tests/text/test_configuration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/text/test_configuration.py -------------------------------------------------------------------------------- /tests/text/test_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/text/test_dataset.py -------------------------------------------------------------------------------- /tests/text/test_features_configuration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/text/test_features_configuration.py -------------------------------------------------------------------------------- /tests/text/test_features_transformers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/text/test_features_transformers.py -------------------------------------------------------------------------------- /tests/text/test_hpo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/text/test_hpo.py -------------------------------------------------------------------------------- /tests/text/test_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/text/test_metrics.py -------------------------------------------------------------------------------- /tests/text/test_model_predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/text/test_model_predict.py -------------------------------------------------------------------------------- /tests/text/test_pipeline_copy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/text/test_pipeline_copy.py -------------------------------------------------------------------------------- /tests/text/test_pipeline_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/text/test_pipeline_datasets.py -------------------------------------------------------------------------------- /tests/text/test_pipeline_predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/text/test_pipeline_predict.py -------------------------------------------------------------------------------- /tests/text/test_pipeline_save.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/text/test_pipeline_save.py -------------------------------------------------------------------------------- /tests/text/test_pipeline_to_mlflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/text/test_pipeline_to_mlflow.py -------------------------------------------------------------------------------- /tests/text/test_pipeline_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/text/test_pipeline_tokenizer.py -------------------------------------------------------------------------------- /tests/text/test_pipeline_vocab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/text/test_pipeline_vocab.py -------------------------------------------------------------------------------- /tests/text/test_pipeline_with_custom_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/text/test_pipeline_with_custom_head.py -------------------------------------------------------------------------------- /tests/text/test_pipeline_with_optional_inputs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/text/test_pipeline_with_optional_inputs.py -------------------------------------------------------------------------------- /tests/text/test_pretrained_word_vectors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/text/test_pretrained_word_vectors.py -------------------------------------------------------------------------------- /tests/text/test_text_cleaning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/text/test_text_cleaning.py -------------------------------------------------------------------------------- /tests/text/test_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/text/test_tokenizer.py -------------------------------------------------------------------------------- /tests/text/test_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/text/test_trainer.py -------------------------------------------------------------------------------- /tests/text_classification_integration_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argilla-io/biome-text/HEAD/tests/text_classification_integration_test.py --------------------------------------------------------------------------------