diff --git a/machine-learn/creating_model_git.py b/machine-learn/creating_model_git.py index 6e9ffde..ce3b077 100644 --- a/machine-learn/creating_model_git.py +++ b/machine-learn/creating_model_git.py @@ -92,8 +92,8 @@ def main(): patience_counter = 0 best_model_path = "/home/runner/work/Singles-Sorter/Singles-Sorter/machine-learn/best_model" n_iter = 45 - batch_size = 64 - drop_size = 0.4 + batch_size = 32 + drop_size = 0.3 iteration_data = {} logger.info("Starting training...") diff --git a/machine-learn/model_metrics.csv b/machine-learn/model_metrics.csv index b4d557c..e24206a 100644 --- a/machine-learn/model_metrics.csv +++ b/machine-learn/model_metrics.csv @@ -8,6 +8,7 @@ custom_ner_model27-4git,0.893175,0.964744,0.831492 custom_ner_model29git,0.892857,0.967742,0.828729 custom_ner_model27-5git,0.892537,0.970779,0.825967 custom_ner_model29-2git,0.888889,0.958466,0.828729 +custom_ner_model29+23git,0.888227,0.964401,0.823204 custom_ner_model27-1git,0.887906,0.952532,0.831492 custom_ner_model26-3git,0.887574,0.955414,0.828729 custom_ner_model26-2git,0.885587,0.958199,0.823204 diff --git a/machine-learn/model_name.txt b/machine-learn/model_name.txt index 552cbf1..f780d55 100644 --- a/machine-learn/model_name.txt +++ b/machine-learn/model_name.txt @@ -1 +1 @@ -custom_ner_model29+23git \ No newline at end of file +custom_ner_model29+23-1git \ No newline at end of file