a | b/model/README.md | ||
---|---|---|---|
1 | # Pre-processing and model definition |
||
2 | |||
3 | Follow the preprocessing steps from writing/preprocessingoverview.png to generate the required base data |
||
4 | |||
5 | - data_loader.py contains the data loader used by pytorch |
||
6 | - net.py contains the definition of the neural network, loss functions and metrics |