Training

For training run jupyter notebook

Note: we use the dataset from this post, so please refer to it for a detailed description and instructions.