Ahmet Yildirim commited on
Commit ·
9eefaf1
1
Parent(s): 474ff3b
- Fix parametername of lemmatisation
Browse files- modeling_humit_tagger.py +11 -8
modeling_humit_tagger.py
CHANGED
|
@@ -974,17 +974,20 @@ class HumitTaggerModel(torch.nn.Module):
|
|
| 974 |
|
| 975 |
self.eval()
|
| 976 |
|
| 977 |
-
if "
|
| 978 |
tag_config["lemmatize"] = False
|
| 979 |
-
|
| 980 |
-
|
| 981 |
-
else:
|
| 982 |
tag_config["lemmatize"] = True
|
| 983 |
-
|
| 984 |
-
|
| 985 |
-
|
| 986 |
-
if "lemmatize" in tag_config and tag_config["lemmatize"]==False:
|
| 987 |
tag_config["lemmatize"] = False
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 988 |
|
| 989 |
if "one_sentence_per_line" not in tag_config:
|
| 990 |
tag_config["one_sentence_per_line"]=False
|
|
|
|
| 974 |
|
| 975 |
self.eval()
|
| 976 |
|
| 977 |
+
if "lemmatisation" in tag_config and tag_config["lemmatisation"]==False:
|
| 978 |
tag_config["lemmatize"] = False
|
| 979 |
+
del tag_config["lemmatisation"]
|
| 980 |
+
elif "lemmatisation" in tag_config:
|
|
|
|
| 981 |
tag_config["lemmatize"] = True
|
| 982 |
+
del tag_config["lemmatisation"]
|
| 983 |
+
elif "lemmatization" in tag_config and tag_config["lemmatization"]==False:
|
|
|
|
|
|
|
| 984 |
tag_config["lemmatize"] = False
|
| 985 |
+
del tag_config["lemmatization"]
|
| 986 |
+
elif "lemmatization" in tag_config:
|
| 987 |
+
tag_config["lemmatize"] = True
|
| 988 |
+
del tag_config["lemmatization"]
|
| 989 |
+
else:
|
| 990 |
+
tag_config["lemmatize"] = True
|
| 991 |
|
| 992 |
if "one_sentence_per_line" not in tag_config:
|
| 993 |
tag_config["one_sentence_per_line"]=False
|