3 [@s:ddlZddlZddlZddlmZejdZddZdS)N)Sessions3cCstjtj|}|d}|d}tj}|jj}dj|}d}d} d} dj||} dj|| } tjj d|d d | d } | j | d d d| j t j d}g}g}xZtdD]N}dj| |}|j||}|jdjjd}|jtj|dddqWd}dj|}dtj||ddS)Nbucketmodelzsagemaker-sample-data-{}z"batch-transform/mnist-1000-samplesz*ServerlessAIWorkshop/BatchTransform/outputz)kmeans_batch_example/input/valid-data.csvz s3://{}/{}zServerless-Workshopzml.c5.18xlarge)Zbase_transform_job_nameZ model_nameZinstance_countZ instance_typeZ output_pathztext/csvZLine) content_typeZ split_typer z{}/data-{}.csv.outBodyzutf-8outputsclassesZint64Valz7, 2, 1, 0, 4, 1, 4, 9, 5, 9z, ) originals predictions) statusCodebody)jsonloadsdumps sagemakerrZ boto_session region_nameformat transformerZ TransformerZ transformwaitboto3resourcerangeZObjectgetreaddecodeextendjoin)eventcontextdataZBUCKETZMODELZsagemaker_sessionregionZ INPUT_BUCKETZDATADIRZOUTPUT input_keyZinput_locationZoutput_locationrrrresultsiZfile_keyZ output_objoutputr r)f/Users/skrinak/Development/serverless-ai-workshop/Lab 2 - SageMaker Batch Transform/lambda_function.pylambda_handler s>        r+)rrrZsagemaker.sessionrrrr+r)r)r)r*s