24 lines (16 loc) · 578 Bytes

Classification models

set up

Go to 'general_config' in utils and set the GPU kernel that you want to use.
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'

data
- training
- testing
model_config (contains files where models are loaded and configurated)
training_config (contains training files for different models)
utils
- general_config
- helper
- preprocessing
- training