train.py -> 1D CNN train_2d.py -> 2D CNN -> better but slower

inference.py -> inference code for 1D CNN inference_2d.py -> inference code for 2D CNN

inference code will save the results, run metrics.py to get the metrics