python - 在 AWS Lambda 中重用 scikit 学习模型 .pkl

Question

我是 AWS 的初学者，我在我的计算机上使用 scikit-learn 训练了一个 RandomForestClassifier，我使用 joblib 以 pkl 格式获取我的模型。

现在我想在 AWS Lambda 中重用这个 RandomForest。因为它需要 sklearn.externals 再次加载我的模型，所以我制作了一个 Zip 目录，其中包含 Numpy、Scipy、sklearn、我的代码和我的 pkl 格式的模型。我将此 zip 放在 S3 存储桶中以在 Lambda 中执行我的代码。

我想知道是否可以使用 model.predict() 使用此模型？我没有找到有关此特定问题的任何文档。

Answer 1

实际上，我成功地使用了 URL。我将我的模型 .pkl 放在 S3 存储桶中，我可以使用 urllib2 获取它。

这是我的代码，如果有人遇到同样的问题:

req = urllib2.Request(url=url_model)
f = urllib2.urlopen(req)
model = cPickle.load(f)

它仅在文件公开时有效。如果你需要它是私有(private)的，你可以像这样使用 boto3 生成一个预签名的 URL:

url_model = s3.generate_presigned_url(
    ClientMethod='get_object',
    ExpiresIn=1,
    Params={
        'Bucket': 'my-bucket',
        'Key': 'mymodel.pkl'
    }
)