如何将数据从S3存储桶加载到Sagemaker jupyter笔记本中以训练模型?

时间:2018-11-27 09:00:28

标签: pandas amazon-web-services amazon-s3 amazon-sagemaker

我在S3存储桶中有csv文件,我想用这些文件在sagemaker中训练模型。

使用此代码,但会出现错误(找不到文件)

import boto3
import pandas as pd
region = boto3.Session().region_name
train_data_location = 's3://taggingu-{}/train.csv'.format(region)
df=pd.read_csv(train_data_location, header = None)
print df.head

对此有什么解决方案?

1 个答案:

答案 0 :(得分:0)

不确定,但是这个stackoverflow可以回答吗? Load S3 Data into AWS SageMaker Notebook

引用@Chhoser:

import boto3
import pandas as pd
from sagemaker import get_execution_role

role = get_execution_role()
bucket='my-bucket'
data_key = 'train.csv'
data_location = 's3://{}/{}'.format(bucket, data_key)

pd.read_csv(data_location)
相关问题