## Todos #20 - [ ] Add training/inference script - [ ] Train the model - [ ] Evaluate results - [ ] Update documentation