A Tensorflow SentenceTransformer layer that takes in strings as input.
Project description
tf-sentence-transformers
A simple tensorflow/keras wrapper around sentence-transformers
. The main class is SentenceTransformer
, a keras.Layer
that can be constructed from a pretrained checkpoint in huggingface model hub.
Example usage:
from tf_sentence_transformers import SentenceTransformer
layer = SentenceTransformer.from_pretrained(
"sentence-transformers/all-MiniLM-L6-v2"
)
inputs = [
["This is a test sentence."],
["This is another test sentence."],
]
embedding = layer(inputs)
# assert embedding.shape == (2, 384)
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Close
Hashes for tf_sentence_transformers-0.1.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 5e550d71eb44f74c7b0ed2d0eb6f9c05abffd03a7ab8631bbcb279f00bb05a15 |
|
MD5 | f3f3d664540e871a7f4a31734148cd67 |
|
BLAKE2b-256 | 0c3098119cbf43cf31c8285e192c51108c02ecfd8bb469e3ca3ae5a80dcf29f9 |
Close
Hashes for tf_sentence_transformers-0.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 59b68b38c5f495ee9f8ab7ab8b97fef04b6d883f48c758c4826be1c464654016 |
|
MD5 | a7f460655155081cbf9bdfd15e2a64cd |
|
BLAKE2b-256 | 8dbc5f756e44a73d1a01c53e057cb1ffab1a7df5d4c7a8750c0dc8af247e05c1 |