在AWS Lambda中重用scikit learn model .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 Bucket中，我可以使用urllib2來獲取它。

這是我的代碼，如果有人遇到過同樣的問題：

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

它僅在文件是公共的時才有效。 如果您需要將其設為私有，則可以使用boto3生成預簽名URL，如下所示：

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