unified embedding model
Project description
uniem
unified embedding model
Install
pip install uniem
Usage
from uniem import UniEmbedder
embedder = UniEmbedder.from_pretrained('uniem/base-softmax-last-mean')
embeddings = embedder.encode(['Hello World!', '你好,世界!'])
Train Your Model
- create virtual environment
conda create -n uniem python=3.10
- install uniem
pip install -e .
- get help
python scripts/train_medi.py --help
- train embedding model
python scripts/train_medi.py <model_path_or_name> <data_file>
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
uniem-0.1.1.tar.gz
(12.7 kB
view hashes)
Built Distribution
uniem-0.1.1-py3-none-any.whl
(14.0 kB
view hashes)