|
--- |
|
pipeline_tag: sentence-similarity |
|
tags: |
|
- sentence-transformers |
|
- feature-extraction |
|
- sentence-similarity |
|
- transformers |
|
language: |
|
- pl |
|
license: cc-by-sa-4.0 |
|
library_name: sentence-transformers |
|
datasets: |
|
- radlab/polish-sts-dataset |
|
models: |
|
- sdadas/polish-roberta-large-v2 |
|
--- |
|
|
|
## Sample model usage |
|
|
|
Below is an example of using the model: |
|
|
|
|
|
```python |
|
from sentence_transformers.cross_encoder import CrossEncoder |
|
|
|
model_path = "radlab/polish-cross-encoder" |
|
model = CrossEncoder(model_path) |
|
|
|
|
|
questions = [ |
|
"Jaką mamy dziś pogodę? bo Andrzej nic nie mówił.", |
|
"Gdzie jedzie Andrzej? Bo wczoraj był w Warszawie.", |
|
"Czy oskarżony się zgadza z przedstawionym wyrokiem?", |
|
] |
|
answers = [ |
|
"Pan Andrzej siedzi w pociągu i jedzie do Wiednia. Ogląda na telefonie zabawne filmiki.", |
|
"Poada deszcz i jest wilgotno, jednak wczoraj było słonecznie.", |
|
"Wyrok jest prawomocny i nie podlega dalszym rozważaniom.", |
|
] |
|
for question in questions: |
|
context_with_question = [(s, question) for s in answers] |
|
results = sorted( |
|
{ |
|
idx: r for idx, r in enumerate(model.predict(context_with_question)) |
|
}.items(), |
|
key=lambda x: x[1], |
|
reverse=True, |
|
) |
|
|
|
print(f"QUESTION: {question}") |
|
print("ANSWERS (sorted):") |
|
for idx, score in results: |
|
print(f"\t[{score}]\t{answers[idx]}") |
|
print("") |
|
``` |
|
|
|
and output to the standard output: |
|
|
|
``` |
|
QUESTION: Jaką mamy dziś pogodę? bo Andrzej nic nie mówił. |
|
ANSWERS (sorted): |
|
[0.016749681904911995] Poada deszcz i jest wilgotno, jednak wczoraj było słonecznie. |
|
[0.01602918468415737] Pan Andrzej siedzi w pociągu i jedzie do Wiednia. Ogląda na telefonie zabawne filmiki. |
|
[0.016013670712709427] Wyrok jest prawomocny i nie podlega dalszym rozważaniom. |
|
|
|
QUESTION: Gdzie jedzie Andrzej? Bo wczoraj był w Warszawie. |
|
ANSWERS (sorted): |
|
[0.5997582674026489] Pan Andrzej siedzi w pociągu i jedzie do Wiednia. Ogląda na telefonie zabawne filmiki. |
|
[0.4528200924396515] Wyrok jest prawomocny i nie podlega dalszym rozważaniom. |
|
[0.17350871860980988] Poada deszcz i jest wilgotno, jednak wczoraj było słonecznie. |
|
|
|
QUESTION: Czy oskarżony się zgadza z przedstawionym wyrokiem? |
|
ANSWERS (sorted): |
|
[0.8431766629219055] Wyrok jest prawomocny i nie podlega dalszym rozważaniom. |
|
[0.6823258996009827] Poada deszcz i jest wilgotno, jednak wczoraj było słonecznie. |
|
[0.558414101600647] Pan Andrzej siedzi w pociągu i jedzie do Wiednia. Ogląda na telefonie zabawne filmiki. |
|
``` |