This example of machine learning uses a linear model for prediction. Is it possible to change the method in SageMaker for this example?
https://github.com/aws/amazon-sag