使用s3fs上传sklearn模型到S3 bucket的问题

Question

我试图使用s3fs将一个SVR模型（用sklearn创建的）上传到S3 bucket，但我得到一个错误信息 "TypeError: a bytes-like object is required, not 'SVR'"。谁能建议如何将SVR转换为正确的格式？

我的代码是

model = SVR_model

fs = s3fs.S3FileSystem()
with fs.open('s3://bucket/SVR_model', 'wb') as f:
    f.write(model)

Answer 1

用pickle把模型变成一个字节对象。

model = pickle.dumps(SVR_model)

fs = s3fs.S3FileSystem()
with fs.open('s3://bucket/SVR_model', 'wb') as f:
    f.write(model)