--- 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. ```