diff options
Diffstat (limited to 'sentence_embedding.py')
-rw-r--r-- | sentence_embedding.py | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/sentence_embedding.py b/sentence_embedding.py index 2ac6720..842fae7 100644 --- a/sentence_embedding.py +++ b/sentence_embedding.py | |||
@@ -103,10 +103,13 @@ def main(args): | |||
103 | 103 | ||
104 | result = zip(row_ind, col_ind) | 104 | result = zip(row_ind, col_ind) |
105 | hit_at_one = len([x for x, y in result if x == y]) | 105 | hit_at_one = len([x for x, y in result if x == y]) |
106 | p_at_one = hit_at_one / instances | ||
106 | percentage = hit_at_one / instances * 100 | 107 | percentage = hit_at_one / instances * 100 |
107 | 108 | ||
108 | if not batch: | 109 | if not batch: |
109 | print(f"{hit_at_one} definitions have been matched correctly") | 110 | print(f"{paradigm} - semb on {source_lang} - {target_lang}") |
111 | print(f"P @ 1: {p_at_one}") | ||
112 | print(f"{percentage} {instances} definitions") | ||
110 | 113 | ||
111 | if batch: | 114 | if batch: |
112 | fields = [ | 115 | fields = [ |
@@ -159,7 +162,7 @@ if __name__ == "__main__": | |||
159 | parser.add_argument( | 162 | parser.add_argument( |
160 | "-n", | 163 | "-n", |
161 | "--instances", | 164 | "--instances", |
162 | help="number of instances in each language to retrieve", | 165 | help="number of instances in each language to use", |
163 | default=1000, | 166 | default=1000, |
164 | type=int, | 167 | type=int, |
165 | ) | 168 | ) |
@@ -167,7 +170,7 @@ if __name__ == "__main__": | |||
167 | "-b", | 170 | "-b", |
168 | "--batch", | 171 | "--batch", |
169 | action="store_true", | 172 | action="store_true", |
170 | help="running in batch (store results in csv) or" | 173 | help="running in batch (store results in csv) or " |
171 | + "running a single instance (output the results)", | 174 | + "running a single instance (output the results)", |
172 | ) | 175 | ) |
173 | parser.add_argument( | 176 | parser.add_argument( |