raphaelsty's picture
Upload model
ab52c51
|
raw
history blame
982 Bytes

semanlink_all_mpnet_base_v2

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.

semanlink_all_mpnet_base_v2 has been fine-tuned on the knowledge graph Semanlink via the library MKB on the link-prediction task. The model is dedicated to the representation of both technical and generic terminology in machine learning, NLP, news.

Usage (Sentence-Transformers)

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 = ["Machine Learning", "Geoffrey Hinton"]

model = SentenceTransformer('raphaelsty/semanlink_all_mpnet_base_v2')
embeddings = model.encode(sentences)
print(embeddings)