From 6f688ff9bacee93679f6af45a301b4308e19764c Mon Sep 17 00:00:00 2001 From: Scott Gasch Date: Wed, 17 Nov 2021 22:14:37 -0800 Subject: WaitableState thingy and some stuff in the ML world. --- ml/model_trainer.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) (limited to 'ml') diff --git a/ml/model_trainer.py b/ml/model_trainer.py index f9e132e..f61b8e7 100644 --- a/ml/model_trainer.py +++ b/ml/model_trainer.py @@ -356,6 +356,13 @@ class TrainingBlueprint(ABC): import input_utils import string_utils + now: datetime.datetime = datetime_utils.now_pacific() + info = f"""Timestamp: {datetime_utils.datetime_to_string(now)} +Model params: {params} +Training examples: {num_examples} +Training set score: {training_score:.2f}% +Testing set score: {test_score:.2f}%""" + print(f'\n{info}\n') if ( (self.spec.persist_percentage_threshold is not None and test_score > self.spec.persist_percentage_threshold) @@ -376,12 +383,6 @@ class TrainingBlueprint(ABC): print(msg) logger.info(msg) model_info_filename = f"{self.spec.basename}_model_info.txt" - now: datetime.datetime = datetime_utils.now_pacific() - info = f"""Timestamp: {datetime_utils.datetime_to_string(now)} -Model params: {params} -Training examples: {num_examples} -Training set score: {training_score:.2f}% -Testing set score: {test_score:.2f}%""" with open(model_info_filename, "w") as f: f.write(info) msg = f"Wrote {model_info_filename}:" -- cgit v1.3