简体   繁体   English

使用 python 从 S3 存储桶下载后加载泡菜文件的问题

[英]Problem with pickle file to load after downloading from S3 bucket using python

I am trying to download my ML model as a pickle file from the S3 bucket and then load it to predict.我正在尝试从 S3 存储桶下载我的 ML model 作为泡菜文件,然后加载它以进行预测。

import boto3
import io
import pickle

REGION = 'us-east-1'
ACCESS_KEY_ID = ******
SECRET_ACCESS_KEY = ******

'BUCKET_NAME = 'aws-logs-914250087788-us-east-1'
KEY = input_name # file path in S3

s3c = boto3.client(
        's3',
        region_name = REGION,
        aws_access_key_id = ACCESS_KEY_ID,
        aws_secret_access_key = SECRET_ACCESS_KEY
    )
rfModel=s3c.download_file(BUCKET_NAME, model1, 'RF_model.pckl')

#load model and predict

lm = open('RF_model.pckl', 'rb')
clf=pickle.loads(lm)
rf_pred=clf.predict(x)
np.savetxt(output_name1, rf_pred, delimiter=',', fmt='%s')

But when I run my code, the following error occurs:但是当我运行我的代码时,会出现以下错误:

TypeError: a byte-like object is required, not _io.BufferedReader. TypeError:需要类似字节的 object,而不是 _io.BufferedReader。

How can I fix this error?我该如何解决这个错误?

I should replace this part:我应该替换这部分:

lm = open('RF_model.pckl', 'rb')
clf=pickle.loads(lm)

with this:有了这个:

with open('RF_model.pckl', 'rb') as file:
    clf = pickle.load(file)

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM