great-ai/example/helper/preprocess.py
András Schmelczer ea54cd9c8c Format
Signed-off-by: András Schmelczer <andras@schmelczer.dev>
2022-04-02 13:57:16 +02:00

10 lines
279 B
Python

import re
from sus.clean import clean
from sus.lemmatize_text import lemmatize_text
def preprocess(text: str) -> str:
cleaned = clean(text, convert_to_ascii=True)
lemmas = [re.sub(r"\d+", "NUM", lemma) for lemma in lemmatize_text(cleaned)]
return " ".join(lemmas)