Using Amazon SageMaker for machine learning