Building the appropriate pipeline to SageMaker for training

[Umbaji001]

• Get the proper data formatting for training sets
• The same for test sets
• Try to build and train the current model on Sagemaker