e5_base_finetuned

Model Details

Model Description

• Model Type: Sentence Transformer

• Maximum Sequence Length: 192 tokens

• Output Dimensionality: 768 dimensions

• Similarity Function: Cosine Similarity

• Supported Modality: Text

Model Sources

• Documentation: Sentence Transformers Documentation
• Repository: Sentence Transformers on GitHub
• Hugging Face: Sentence Transformers on Hugging Face

Full Model Architecture

SentenceTransformer(
  (0): Transformer({'transformer_task': 'feature-extraction', 'modality_config': {'text': {'method': 'forward', 'method_output_name': 'last_hidden_state'}}, 'module_output_name': 'token_embeddings', 'architecture': 'XLMRobertaModel'})
  (1): Pooling({'embedding_dimension': 768, 'pooling_mode': 'mean', 'include_prompt': True})
  (2): Normalize({})
)

Usage

Direct Usage (Sentence Transformers)

First install the Sentence Transformers library:

pip install -U sentence-transformers

Then you can load this model and run inference.

from sentence_transformers import SentenceTransformer

# Download from the 🤗 Hub
model = SentenceTransformer("sentence_transformers_model_id")
# Run inference
queries = [
    'query: thức ăn chó pedigree vị gà',
]
documents = [
    'passage: Thức Ăn Chó Trưởng Thành Pedigree Vị Gà & Rau Củ Túi 1.5Kg Thức Ăn Chó Trưởng Thành Pedigree Vị Gà & Rau Củ Túi 1.5Kg là lựa chọn hoàn hảo để chăm sóc sức khỏe cho chú chó yêu của bạn. Sản phẩm kết hợp hài hòa giữa hương vị gà thơm ngon và rau củ tươi mát, cung cấp nguồn protein thô tối thiểu 18%, béo thô tối thiểu 9% cùng các vitamin và khoáng chất thiết yếu. Công thức đặc biệt giúp hỗ trợ hệ tiêu hóa khỏe mạnh, bộ xương vững chắc và bộ lông óng mượt. Thích hợp cho chó từ 18 tháng tuổi trở lên, sản phẩm đóng gói 1.5kg tiện lợi, dễ dàng bảo quản nơi khô ráo, thoáng mát. Pedigree cam kết mang đến dinh dưỡng tối ưu cho mọi giai đoạn phát triển của chó.',
    'passage: ',
    'passage: Bàn máy tính SD01 Bàn máy tính SD01 của Nội Thất Hòa Phát là sự lựa chọn thông minh cho mọi không gian làm việc. Sản phẩm được chế tác từ sự kết hợp hài hòa giữa gỗ công nghiệp và gỗ tự nhiên, mang lại vẻ đẹp tự nhiên cùng độ bền vượt trội. Thiết kế của bàn SD01 chú trọng đến sự tiện dụng và tính thẩm mỹ, giúp bạn dễ dàng sắp xếp máy tính và các vật dụng cần thiết một cách gọn gàng. Đây là giải pháp lý tưởng cho những ai tìm kiếm một chiếc bàn máy tính bền đẹp, phù hợp với nhiều phong cách nội thất khác nhau, từ hiện đại đến truyền thống.',
]
query_embeddings = model.encode_query(queries)
document_embeddings = model.encode_document(documents)
print(query_embeddings.shape, document_embeddings.shape)
# [1, 768] [3, 768]

# Get the similarity scores for the embeddings
similarities = model.similarity(query_embeddings, document_embeddings)
print(similarities)
# tensor([[ 0.7672, -0.0050, -0.0058]])

Training Details

Training Dataset

Unnamed Dataset

• Size: 36,389 training samples
• Columns: query, positive, and negative
• Approximate statistics based on the first 1000 samples:

  	query	positive	negative
  type	string	string	string
  details	min: 8 tokensmean: 11.23 tokensmax: 18 tokens	min: 121 tokensmean: 166.44 tokensmax: 192 tokens	min: 4 tokensmean: 147.22 tokensmax: 192 tokens

• Samples: | query | positive | negative | |:----------------------------------------------|:-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------