Views
No views yet
sentence-transformer library.1git clone https://github.com/HKUNLP/instructor-embedding
2cd sentence-transformers
3pip install -e .1from sentence_transformers import SentenceTransformer
2sentence = "3D ActionSLAM: wearable person tracking in multi-floor environments"
3instruction = "Represent the Science title; Input:"
4model = SentenceTransformer('hku-nlp/instructor-large')
5embeddings = model.encode([[instruction,sentence,0]])
6print(embeddings)1from sklearn.metrics.pairwise import cosine_similarity
2sentences_a = [['Represent the Science sentence; Input: ','Parton energy loss in QCD matter',0],
3 ['Represent the Financial statement; Input: ','The Federal Reserve on Wednesday raised its benchmark interest rate.',0]
4sentences_b = [['Represent the Science sentence; Input: ','The Chiral Phase Transition in Dissipative Dynamics', 0],
5 ['Represent the Financial statement; Input: ','The funds rose less than 0.5 per cent on Friday',0]
6embeddings_a = model.encode(sentences_a)
7embeddings_b = model.encode(sentences_b)
8similarities = cosine_similarity(embeddings_a,embeddings_b)
9print(similarities)