Using the SageMaker training toolkit with scikit-learn