Huggingface
HuggingFaceEmbedding
Bases: AbstractEmbeddingModel
Class for embedding models using HuggingFace.
Source code in src/embedding/huggingface.py
9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
|
get_embedding(text)
Returns the embedding of the text.
Parameters:
Name | Type | Description | Default |
---|---|---|---|
text |
str
|
|
required |
Returns:
Type | Description |
---|---|
ndarray
|
|
Source code in src/embedding/huggingface.py
21 22 23 24 25 26 27 28 29 30 31 32 |
|
SentenceTransformersEmbedding
Bases: AbstractEmbeddingModel
Source code in src/embedding/huggingface.py
35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 |
|
get_embedding(text)
Returns the embedding of the text.
Parameters:
Name | Type | Description | Default |
---|---|---|---|
text |
str
|
|
required |
Returns:
Type | Description |
---|---|
ndarray
|
|
Source code in src/embedding/huggingface.py
42 43 44 45 46 47 48 49 50 51 |
|