Browse Source
Update readme
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
1 changed files with
14 additions and
15 deletions
-
README.md
|
|
@ -1,6 +1,6 @@ |
|
|
# Text Embedding with Transformers |
|
|
# Text Embedding with Transformers |
|
|
|
|
|
|
|
|
*author: Jael Gu* |
|
|
|
|
|
|
|
|
*author: [Jael Gu](https://github.com/jaelgu)* |
|
|
|
|
|
|
|
|
<br /> |
|
|
<br /> |
|
|
|
|
|
|
|
|
@ -38,10 +38,10 @@ Create the operator via the following factory method |
|
|
|
|
|
|
|
|
***model_name***: *str* |
|
|
***model_name***: *str* |
|
|
|
|
|
|
|
|
The model name in string. |
|
|
|
|
|
|
|
|
The model name in string. |
|
|
The default model name is "bert-base-uncased". |
|
|
The default model name is "bert-base-uncased". |
|
|
|
|
|
|
|
|
Supported model names: |
|
|
|
|
|
|
|
|
Supported model names: |
|
|
|
|
|
|
|
|
<details><summary>Albert</summary> |
|
|
<details><summary>Albert</summary> |
|
|
|
|
|
|
|
|
@ -303,12 +303,11 @@ and then return text embedding in ndarray. |
|
|
|
|
|
|
|
|
***text***: *str* |
|
|
***text***: *str* |
|
|
|
|
|
|
|
|
The text in string. |
|
|
|
|
|
|
|
|
The text in string. |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
**Returns**: |
|
|
**Returns**: |
|
|
|
|
|
|
|
|
*numpy.ndarray* |
|
|
*numpy.ndarray* |
|
|
|
|
|
|
|
|
The text embedding extracted by model. |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
The text embedding extracted by model. |
|
|
|
