TDApiClient exposes the following classes:
- Two estimator wrapper classes (returned by TDApiClient.<sagemaker_class_name>)
- One predictor wrapper classes (returned by deploy method)
The syntax is:
estimatorWrapper.cloudObj
predictoWrapper.cloudObj
Example
from tdapiclient import create_tdapi_context, TDApiClient
awsContext = create_tdapi_context("aws", "s3_bucket")
sagemakerTD = TDApiClient(awsContext)
mxnetEstimaterWrapperObj = sagemakerTD.MXNet('train.py', role='SageMakerRole', instance_type='ml.p2.xlarge', instance_count=1, py_version="py3", framework_version='1.2.1')
mxnetEstimaterWrapperObj.cloudObj
inputDF = DataFrame("inputTable") mxnetEstimaterWrapperObj.fit(inputs=inputDF, content_type='json')
predictorWrapperObj = mxnetEstimaterWrapperObj.deploy(sagemaker_kw_args={"instance_type": "ml.p2.xlarge", "initial_instance_count": 1})
predictorWrapperObj.cloudObj