Ahmet Yildirim commited on
Commit
9eefaf1
·
1 Parent(s): 474ff3b

- Fix parametername of lemmatisation

Browse files
Files changed (1) hide show
  1. 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 "lemmatise" in tag_config and tag_config["lemmatise"]==False:
978
  tag_config["lemmatize"] = False
979
- if "lemmatise" in tag_config:
980
- del tag_config["lemmatise"]
981
- else:
982
  tag_config["lemmatize"] = True
983
- if "lemmatise" in tag_config:
984
- del tag_config["lemmatise"]
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