├── .github └── ISSUE_TEMPLATE │ ├── bug_report.md │ └── feature_request.md ├── .gitignore ├── LICENSE ├── Makefile ├── README.md ├── ensemble_transformers ├── __init__.py ├── __version__.py ├── base.py ├── config.py ├── ensemble.py └── output.py ├── pyproject.toml ├── requirements-dev.txt ├── requirements.txt ├── setup.cfg ├── setup.py └── tests ├── __init__.py ├── test_basic.py └── test_invalid.py /.github/ISSUE_TEMPLATE/bug_report.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaketae/ensemble-transformers/HEAD/.github/ISSUE_TEMPLATE/bug_report.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature_request.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaketae/ensemble-transformers/HEAD/.github/ISSUE_TEMPLATE/feature_request.md -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaketae/ensemble-transformers/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaketae/ensemble-transformers/HEAD/LICENSE -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaketae/ensemble-transformers/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaketae/ensemble-transformers/HEAD/README.md -------------------------------------------------------------------------------- /ensemble_transformers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaketae/ensemble-transformers/HEAD/ensemble_transformers/__init__.py -------------------------------------------------------------------------------- /ensemble_transformers/__version__.py: -------------------------------------------------------------------------------- 1 | version = "0.0.2" 2 | -------------------------------------------------------------------------------- /ensemble_transformers/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaketae/ensemble-transformers/HEAD/ensemble_transformers/base.py -------------------------------------------------------------------------------- /ensemble_transformers/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaketae/ensemble-transformers/HEAD/ensemble_transformers/config.py -------------------------------------------------------------------------------- /ensemble_transformers/ensemble.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaketae/ensemble-transformers/HEAD/ensemble_transformers/ensemble.py -------------------------------------------------------------------------------- /ensemble_transformers/output.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaketae/ensemble-transformers/HEAD/ensemble_transformers/output.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- 1 | [tool.black] 2 | line-length = 119 3 | target-version = ['py37'] -------------------------------------------------------------------------------- /requirements-dev.txt: -------------------------------------------------------------------------------- 1 | -r requirements.txt 2 | black 3 | flake8 4 | isort 5 | pytest -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | torch 2 | transformers -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaketae/ensemble-transformers/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaketae/ensemble-transformers/HEAD/setup.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/test_basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaketae/ensemble-transformers/HEAD/tests/test_basic.py -------------------------------------------------------------------------------- /tests/test_invalid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaketae/ensemble-transformers/HEAD/tests/test_invalid.py --------------------------------------------------------------------------------