首页 > 解决方案 > AWS - 是否可以将参数从 AWS lambda 函数传递到 AWS sagemaker notebook

问题描述

我是 AWS 控制台的初学者。我在构建机器学习管道时遇到问题。

目前,Lambda 函数负责从前端获取上传的文件名、用户名和调用笔记本实例。

此外,实例的生命周期配置将调用笔记本进行训练。所以问题是如何将上述变量变量传递给 sagemaker notebook 以训练机器学习模型。有可能实现这一目标吗?谢谢你。

#invoke command from lambda
client.start_notebook_instance(NotebookInstanceName='<sagemaker_instance_name>')

sagemaker实例下的生命周期配置:

#!/bin/bash

set -e

ENVIRONMENT=JupyterSystemEnv
#JupyterSystemEnv
NOTEBOOK_FILE=/home/ec2-user/SageMaker/XGBoost_training.ipynb

source /home/ec2-user/anaconda3/bin/activate "$ENVIRONMENT"

nohup jupyter nbconvert --ExecutePreprocessor.timeout=-1 --ExecutePreprocessor.kernel_name=python3 --to notebook --execute "$NOTEBOOK_FILE" &

标签: amazon-web-servicesaws-lambdaamazon-sagemaker

解决方案


推荐阅读