Follow the steps below to train your own model:
- Preprocess your dataset using
preprocess.py - Split your data to train, test and validation sets and run
train.py
Make sure the name and column names of your datasets match with the names in the training code.
Don't forget to use GPU for trainig :)