From 498d93de2aeabe8a2f01103e1bb55f7d91bc7672 Mon Sep 17 00:00:00 2001 From: coolneng Date: Thu, 10 Jun 2021 13:27:55 +0200 Subject: [PATCH] Execute the training loop in the model module --- src/model.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/model.py b/src/model.py index 9530d8a..94ee44a 100644 --- a/src/model.py +++ b/src/model.py @@ -85,3 +85,7 @@ def run(data_file, label_file, seed_value=42) -> None: ) print("Training complete. Obtaining final metrics...") show_metrics(model, eval_data, test_data) + + +if __name__ == "__main__": + run(data_file="data/curesim-HVR.fastq", label_file="data/HVR.fastq")