andras/great-ai
1
0
Fork 0
Code Issues Pull requests 1 Projects Releases 4 Packages Wiki Activity Actions

Improve example

Browse source
This commit is contained in:
Andras Schmelczer 2022-05-26 21:49:58 +02:00
parent e4b4c262c3
commit 8033e63b51
2 changed files with 20 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

2
examples/simple/helpers.py
View file

@ -12,5 +12,5 @@ def preprocess(text: str) -> str:
def lemmatize(text: str) -> str:
lemmatized = lemmatize_text(text)
clean_lemmas = [re.sub(r"\d[\d.,]*", "NUM", lemma) for lemma in lemmatized]
clean_lemmas = [re.sub(r"\d[\d.,]*", "NUM", lemma.lower()) for lemma in lemmatized]
return " ".join(clean_lemmas)