ניסיון 400 איטרציות

NHLOCAL · Aug 1, 2023 · 1e4a516 · 1e4a516
1 parent b5bcbd6
commit 1e4a516
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/machine-learn/create_ner_model.py b/machine-learn/create_ner_model.py
@@ -56,11 +56,11 @@ def convert_to_spacy_format(data):
 nlp.begin_training()
 
 # Training loop
-for itn in range(1000):
+for itn in range(400):
     losses = {}
     for example in training_data:
         nlp.update([example], drop=0.35, losses=losses)
-    if int(losses['ner']) <= 100: break
+    if int(losses['ner']) <= 150: break
     print(str(itn) + ": " + str(losses))