CallEstimator

estimator.py

# Call the estimator with the main parameters

from sagemaker.estimator import Estimator

estimator = Estimator(
    image_uri=ecr_image,
    role=role,
    instance_count=1,
    instance_type="ml.c4.xlarge",
    output_path=out_path
)

hyperparameters = {
}

print(hyperparameters)
estimator.set_hyperparameters(**hyperparameters)


# Train the model
estimator.fit({'train': train_path})