pandas - 如何将数据从 S3 存储桶加载到 Sagemaker jupyter 笔记本来训练模型？

Question

我在 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

有什么办法可以解决这个问题吗？

Answer 1

不确定，但是这个 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)