krotima1
commited on
Commit
·
de43f74
1
Parent(s):
bbf50b1
feat: add final AlignScoreCS.py file for ease of use with transformers.
Browse files- AlignScoreCS.py +2 -2
AlignScoreCS.py
CHANGED
|
@@ -6,7 +6,7 @@ import numpy as np
|
|
| 6 |
from typing import TYPE_CHECKING, Any, Callable, Dict, List, Optional, Tuple, Union
|
| 7 |
import torch.nn as nn
|
| 8 |
import torch
|
| 9 |
-
from nltk.tokenize import sent_tokenize #If you don't have nltk, you can use another sentence tokenizer
|
| 10 |
from tqdm import tqdm
|
| 11 |
|
| 12 |
class AlignScoreCS(transformers.XLMRobertaModel):
|
|
@@ -456,4 +456,4 @@ class AlignScoreCS(transformers.XLMRobertaModel):
|
|
| 456 |
, ("Jaký má číslo drezu Jaromír Jágr?", "Jaromír Jágr má číslo drezu 68."), ("Kolik je Jaromíru Jágrovi let?", "Jaromíru Jágrovi je 68."), ("Kolik je Jaromíru Jágrovi let?", "Jaromíru Jágrovi je 67.")
|
| 457 |
, ("Co udělali lidi, když Jágr střelil gól?", "Lidi začali šílet. Dokonce šílely i po zápase."), ("Co udělali lidi, když Jágr střelil gól?", "Šli dát góla Haškovi")]
|
| 458 |
for co,cl in zip(contexts*len(claims),claims):
|
| 459 |
-
print(round(
|
|
|
|
| 6 |
from typing import TYPE_CHECKING, Any, Callable, Dict, List, Optional, Tuple, Union
|
| 7 |
import torch.nn as nn
|
| 8 |
import torch
|
| 9 |
+
from nltk.tokenize import sent_tokenize #If you don't have nltk, you can use another sentence tokenizer (morphodita, spacy, etc.)
|
| 10 |
from tqdm import tqdm
|
| 11 |
|
| 12 |
class AlignScoreCS(transformers.XLMRobertaModel):
|
|
|
|
| 456 |
, ("Jaký má číslo drezu Jaromír Jágr?", "Jaromír Jágr má číslo drezu 68."), ("Kolik je Jaromíru Jágrovi let?", "Jaromíru Jágrovi je 68."), ("Kolik je Jaromíru Jágrovi let?", "Jaromíru Jágrovi je 67.")
|
| 457 |
, ("Co udělali lidi, když Jágr střelil gól?", "Lidi začali šílet. Dokonce šílely i po zápase."), ("Co udělali lidi, když Jágr střelil gól?", "Šli dát góla Haškovi")]
|
| 458 |
for co,cl in zip(contexts*len(claims),claims):
|
| 459 |
+
print(round(self.score(co, cl[1],eval_mode="nli_sp",eval_question_answer=cl[0] ).tolist()[0],5)," ".join(cl))
|