Hugging Face's logo

cambridgeltl
/
SapBERT-from-PubMedBERT-fulltext

Feature Extraction
Transformers
PyTorch
TensorFlow
JAX
Safetensors
English
bert
biomedical
lexical semantics
bionlp
biology
science
embedding
entity linking
Model card Files
xet
 Community
3

Expected input for the hosted inference API

#1
by jdsouza - opened
Discussion
jdsouza

Dear authors,

what is the input expected for the hosted inference API of your model? And how do we interpret the resulting output dimensions?

Many thanks for your support in advance.
fl399
 Language Technology Lab @University of Cambridge org

Hi,

We have added the following information:

The input should be a string of biomedical entity names, e.g., "covid infection" or "Hydroxychloroquine". The [CLS] embedding of the last layer is regarded as the output.

Thanks!
fl399 changed discussion status to closed

Sign up or log in to comment