.gitignore
.pre-commit-config.yaml
.readthedocs.yaml
LICENSE
README.md
example.py
pyproject.toml
requirements.txt
train_model.py
dataset2vec/__init__.py
dataset2vec/config.py
dataset2vec/data.py
dataset2vec/model.py
dataset2vec/train.py
dataset2vec/utils.py
dataset2vec.egg-info/PKG-INFO
dataset2vec.egg-info/SOURCES.txt
dataset2vec.egg-info/dependency_links.txt
dataset2vec.egg-info/requires.txt
dataset2vec.egg-info/top_level.txt
docs/Makefile
docs/modules.rst
docs/source/conf.py
docs/source/index.rst
docs/source/modules/config.rst
docs/source/modules/data.rst
docs/source/modules/model.rst
scripts/check_code.sh
test/conftest.py
test/dataset2vec/test_data.py
test/dataset2vec/test_model.py
test/dataset2vec/test_train.py
test/dataset2vec/test_utils.py
test/integration/test_model_training.py