happy-wolf-7852
06/02/2023, 4:20 PMancient-application-36103
06/02/2023, 4:26 PMhappy-wolf-7852
06/02/2023, 4:40 PM@batch
@step
def trigger_batch_transform(self):
import sagemaker
from sagemaker.transformer import Transformer
score_files = aws_s3.get_files_in_folder(self.s3_bucket_name, self.score_sub_folder_name)
for score_file in score_files:
sagemaker_session = sagemaker.Session()
input_data = f's3://{self.s3_bucket_name}/{score_file}'
output_path = f's3://{self.s3_bucket_name}/{self.score_results_sub_folder_name}'
transformer = Transformer(
model_name=self.model_name,
instance_count=1,
instance_type='ml.m5.4xlarge',
strategy='MultiRecord',
assemble_with='Line',
output_path=output_path,
base_transform_job_name='batch-transform-job',
sagemaker_session=sagemaker_session,
accept='text/csv'
)
transformer.transform(
data=input_data,
data_type='S3Prefix',
content_type='text/csv',
split_type='Line'
)
transformer.wait()
self.next(self.end)ancient-application-36103
06/02/2023, 5:05 PMancient-application-36103
06/02/2023, 5:06 PMhappy-wolf-7852
06/02/2023, 7:55 PMancient-application-36103
06/02/2023, 7:59 PMhappy-wolf-7852
06/03/2023, 10:16 AMhappy-wolf-7852
06/03/2023, 6:24 PMancient-application-36103
06/04/2023, 3:24 AM