Compare commits

..

2 Commits

Author SHA1 Message Date
coolneng 197c532113
Add usage instructions 2021-06-10 13:28:32 +02:00
coolneng 498d93de2a
Execute the training loop in the model module 2021-06-10 13:27:55 +02:00
2 changed files with 12 additions and 0 deletions

View File

@ -39,3 +39,11 @@ nix-shell
After running these commands, you will find yourself in a shell that
contains all the needed dependencies.
## Usage
The following creates the dataset, trains the Deep Learning model and shows the accuracy:
```bash
poetry run python src/model.py
```

View File

@ -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")