Commit 99c7e165 authored by Tomasz Walkowiak's avatar Tomasz Walkowiak

sentences

parent a3e35999
Pipeline #2564 failed with stages
in 30 seconds
......@@ -43,7 +43,7 @@ class Model:
self.synsets[str(synset)] = (tag, lemma)
else:
if var is not None:
self.variants[lemma + "." + str(var)] = (tag, lemma)
self.variants[lemma.replace(" ","_") + "." + str(var)] = (tag, lemma)
else:
self.lemmas[lemma] = tag
wb.close()
......@@ -124,7 +124,7 @@ class Category:
el = el.split("(")[0]
if (el in model.variants and
model.variants[el][1] == base):
self.add(stat, model.variants[el][0], id,
self.add(stat, model.variants[el][0], el,
sentence)
self.inc(stat, model.variants[el][0])
if self.verbose:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment