a | b/src/train.py | ||
---|---|---|---|
1 | #!/usr/bin/env python |
||
2 | from compnet import train_model |
||
3 | from settings import Settings |
||
4 | |||
5 | |||
6 | if __name__ == '__main__': |
||
7 | |||
8 | settings = Settings() |
||
9 | |||
10 | data_params = settings['DATA'] |
||
11 | train_params = settings['TRAINING'] |
||
12 | common_params = settings['COMMON'] |
||
13 | |||
14 | train_model(data_params, train_params, common_params) |