Execute the training loop in the model module

This commit is contained in:
coolneng 2021-06-10 13:27:55 +02:00
parent 3b2b6c4af9
commit 498d93de2a
Signed by: coolneng
GPG Key ID: 9893DA236405AF57
1 changed files with 4 additions and 0 deletions

View File

@ -85,3 +85,7 @@ def run(data_file, label_file, seed_value=42) -> None:
) )
print("Training complete. Obtaining final metrics...") print("Training complete. Obtaining final metrics...")
show_metrics(model, eval_data, test_data) show_metrics(model, eval_data, test_data)
if __name__ == "__main__":
run(data_file="data/curesim-HVR.fastq", label_file="data/HVR.fastq")