diff --git a/machine-learn/creating_model_git.py b/machine-learn/creating_model_git.py index 6e11ed73..c258f349 100644 --- a/machine-learn/creating_model_git.py +++ b/machine-learn/creating_model_git.py @@ -88,7 +88,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.35 + drop_size = 0.4 iteration_data = {} logger.info("Starting training...") diff --git a/machine-learn/model_metrics.csv b/machine-learn/model_metrics.csv index dd4cd630..2280df56 100644 --- a/machine-learn/model_metrics.csv +++ b/machine-learn/model_metrics.csv @@ -12,5 +12,6 @@ custom_ner_model26-2git,0.885587,0.958199,0.823204 custom_ner_model27-2git,0.885587,0.958199,0.823204 custom_ner_model27git,0.883309,0.949206,0.825967 custom_ner_model27-3git,0.881752,0.934985,0.834254 +custom_ner_model28git,0.877246,0.957516,0.809392 custom_ner_model26git,0.876672,0.948553,0.814917 custom_ner_model26-1git,0.873700,0.945338,0.812155 diff --git a/machine-learn/model_name.txt b/machine-learn/model_name.txt index c0b9ef72..d50f6437 100644 --- a/machine-learn/model_name.txt +++ b/machine-learn/model_name.txt @@ -1 +1 @@ -custom_ner_model28git \ No newline at end of file +custom_ner_model28-1git \ No newline at end of file