Add files

Signed-off-by: András Schmelczer <andras@schmelczer.dev>

good_ai/src/sus/lemmatize_token.py

@@ -0,0 +1,20 @@
+from spacy.tokens import Token
+
+from .data import american_spellings
+
+
+def lemmatize_token(
+    token: Token,
+    add_negation: bool = False,
+    add_part_of_speech: bool = False,
+) -> str:
+    lemma = token.lemma_.lower()
+
+    lemma = american_spellings.get(lemma, lemma)
+
+    if add_part_of_speech:
+        lemma = f"{lemma}_{token.pos_}"
+    if add_negation and token._.negex:
+        lemma = f"NOT_{lemma}"
+
+    return lemma