Classification models

set up

1. Go to 'general_config' in utils and set the GPU kernel that you want to use.

2. Within 'training_config', select a model that you want to train. Set the model parameters within 'your_model_train' and run the script.

folder structure

TODO: create automatically with 'tree'

1. data
   • training
   • testing
2. model_config (contains files where models are loaded and configurated)
3. training_config (contains training files for different models)
4. utils
   • general_config
   • helper
   • preprocessing
   • training