scikit-learn>=1.0.0
pandas>=1.0.0
diskcache>=5.6.1
skops>=0.8.0
model2vec

[all]
scikit-learn>=1.0.0
pandas>=1.0.0
diskcache>=5.6.1
skops>=0.8.0
model2vec
sense2vec==2.0.0
bpemb>=0.3.3
gensim>=4.3.1
scipy<1.13.0
timm>=0.6.7
openai>=1.59.8

[bpemb]
bpemb>=0.3.3
scikit-learn>=1.0.0
pandas>=1.0.0
diskcache>=5.6.1
skops>=0.8.0
model2vec

[cohere]
cohere>=4.11.2
scikit-learn>=1.0.0
pandas>=1.0.0
diskcache>=5.6.1
skops>=0.8.0
model2vec

[dev]
scikit-learn>=1.0.0
pandas>=1.0.0
diskcache>=5.6.1
skops>=0.8.0
model2vec
sense2vec==2.0.0
bpemb>=0.3.3
gensim>=4.3.1
scipy<1.13.0
timm>=0.6.7
openai>=1.59.8
mkdocs-material==9.6.9
mkdocstrings==0.29.0
mkdocstrings-python==1.16.0
mktestdocs==0.2.4
interrogate>=1.5.0
pytest>=4.0.2
ruff
pre-commit>=2.2.0
datasets==2.8.0
matplotlib==3.4.3
pytest-xdist

[docs]
mkdocs-material==9.6.9
mkdocstrings==0.29.0
mkdocstrings-python==1.16.0
mktestdocs==0.2.4

[gensim]
gensim>=4.3.1
scipy<1.13.0
scikit-learn>=1.0.0
pandas>=1.0.0
diskcache>=5.6.1
skops>=0.8.0
model2vec

[openai]
openai>=1.59.8
scikit-learn>=1.0.0
pandas>=1.0.0
diskcache>=5.6.1
skops>=0.8.0
model2vec

[pytorch]
torch>=1.12.0
scikit-learn>=1.0.0
pandas>=1.0.0
diskcache>=5.6.1
skops>=0.8.0
model2vec

[sbert]
sentence-transformers>=2.2.2
scikit-learn>=1.0.0
pandas>=1.0.0
diskcache>=5.6.1
skops>=0.8.0
model2vec

[sense2vec]
sense2vec==2.0.0
scikit-learn>=1.0.0
pandas>=1.0.0
diskcache>=5.6.1
skops>=0.8.0
model2vec

[spacy]
spacy>=3.5.0
scikit-learn>=1.0.0
pandas>=1.0.0
diskcache>=5.6.1
skops>=0.8.0
model2vec

[text]
sense2vec==2.0.0
bpemb>=0.3.3
gensim>=4.3.1
scipy<1.13.0
scikit-learn>=1.0.0
pandas>=1.0.0
diskcache>=5.6.1
skops>=0.8.0
model2vec

[vision]
timm>=0.6.7
scikit-learn>=1.0.0
pandas>=1.0.0
diskcache>=5.6.1
skops>=0.8.0
model2vec
