Update create_ner_model_git_json.py

NHLOCAL · Jul 6, 2024 · 570613c · 570613c
1 parent b368653
commit 570613c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/machine-learn/create_ner_model_git_json.py b/machine-learn/create_ner_model_git_json.py
@@ -89,7 +89,7 @@ def custom_tokenizer(nlp):
     for example in training_data:
         nlp.update([example], drop=0.5, losses=losses)
     print(str(itn) + ": " + str(losses))
-    if int(losses['ner']) <= 200:
+    if int(losses['ner']) <= 50:
         break
 
 # Save the trained model to disk