diff --git a/machine-learn/creating_model_git.py b/machine-learn/creating_model_git.py index 76321c6..4e2db34 100644 --- a/machine-learn/creating_model_git.py +++ b/machine-learn/creating_model_git.py @@ -93,7 +93,7 @@ def main(): best_model_path = "/home/runner/work/Singles-Sorter/Singles-Sorter/machine-learn/best_model" n_iter = 100 batch_size = 64 - drop_size = 0.5 + drop_size = 0.45 iteration_data = {} logger.info("Starting training...") diff --git a/machine-learn/model_metrics.csv b/machine-learn/model_metrics.csv index db87df8..aa06e91 100644 --- a/machine-learn/model_metrics.csv +++ b/machine-learn/model_metrics.csv @@ -3,6 +3,7 @@ custom_ner_model23-4git,0.896652,0.971609,0.832432 custom_ner_model23-1git,0.895742,0.980707,0.824324 custom_ner_model25-1git,0.895044,0.971519,0.829730 custom_ner_model24git,0.893119,0.974441,0.824324 +custom_ner_model29+23.4git,0.893119,0.974441,0.824324 custom_ner_model27-5git,0.892805,0.977492,0.821622 custom_ner_model23git,0.892489,0.980583,0.818919 custom_ner_model28git,0.891813,0.971338,0.824324 diff --git a/machine-learn/model_name.txt b/machine-learn/model_name.txt index e29b8c5..82d55a8 100644 --- a/machine-learn/model_name.txt +++ b/machine-learn/model_name.txt @@ -1 +1 @@ -custom_ner_model29+23.4git \ No newline at end of file +custom_ner_model29+23.4-1git \ No newline at end of file