JoBeer/eclassTrainST
Viewer • Updated • 1.15M • 73
How to use JoBeer/all-mpnet-base-v2-eclass with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("JoBeer/all-mpnet-base-v2-eclass")
sentences = [
"That is a happy person",
"That is a happy dog",
"That is a very happy person",
"Today is a sunny day"
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4]How to use JoBeer/all-mpnet-base-v2-eclass with Transformers:
# Load model directly
from transformers import AutoTokenizer, AutoModel
tokenizer = AutoTokenizer.from_pretrained("JoBeer/all-mpnet-base-v2-eclass")
model = AutoModel.from_pretrained("JoBeer/all-mpnet-base-v2-eclass")This is a sentence-transformers model: It maps sentences & paragraphs to a 768 dimensional dense vector space and can be used for tasks like clustering or semantic search.
Using this model becomes easy when you have sentence-transformers installed:
pip install -U sentence-transformers
Then you can use the model like this:
from sentence_transformers import SentenceTransformer
sentences = ["This is an example sentence", "Each sentence is converted"]
model = SentenceTransformer('JoBeer/all-mpnet-base-v2-eclass')
embeddings = model.encode(sentences)
print(embeddings)
For an automated evaluation of this model, see the Sentence Embeddings Benchmark: https://seb.sbert.net
The Model was trained with the eclass-dataset (https://huggingface.co/datasets/JoBeer/eclassTrainST).
SentenceTransformer(
(0): Transformer({'max_seq_length': 384, 'do_lower_case': False}) with Transformer model: MPNetModel
(1): Pooling({'word_embedding_dimension': 768, 'pooling_mode_cls_token': False, 'pooling_mode_mean_tokens': True, 'pooling_mode_max_tokens': False, 'pooling_mode_mean_sqrt_len_tokens': False})
(2): Normalize()
)
from sentence_transformers import SentenceTransformer model = SentenceTransformer("JoBeer/all-mpnet-base-v2-eclass") sentences = [ "That is a happy person", "That is a happy dog", "That is a very happy person", "Today is a sunny day" ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4]