mymusise / sentence-transformers-tf

sentence-transformers with tensorflow

Sentence-Transformers-TF:

A unofficial TensorFlow Implementation of SBERT

Inference

from sentence_transformers_tf.TFSentenceTransformer import TFSentenceTransformer

tfsentent = TFSentenceTransformer("sentence-transformers/stsb-xlm-r-multilingual")

outs = tfsentent.encode(["hi, there", "tihs is a tensorflow implementation of the sbert"])
print(outs.shape)

Finetune

Here is an example: examples/finetune.ipynb

Reference

• Origin repository: https://github.com/UKPLab/sentence-transformers
• Original Paper : https://arxiv.org/abs/1908.10084