krotima1 commited on
Commit
de43f74
·
1 Parent(s): bbf50b1

feat: add final AlignScoreCS.py file for ease of use with transformers.

Browse files
Files changed (1) hide show
  1. 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(model.score(co, cl[1],eval_mode="nli_sp",eval_question_answer=cl[0] ).tolist()[0],5)," ".join(cl))
 
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))