Проблема при загрузке модели sklearn в корзину S3 с использованием s3fs - PullRequest
Новая
       66

Проблема при загрузке модели sklearn в корзину S3 с использованием s3fs

0 голосов
/ 27 апреля 2020

Я пытаюсь загрузить модель SVR (созданную с помощью sklearn) в корзину S3, используя s3fs, но я получаю сообщение об ошибке «TypeError: требуется объект, похожий на байты, а не SVR». Кто-нибудь может подсказать, как преобразовать SVR в нужный формат?

Мой код 

model = SVR_model

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

1 Ответ

1 голос
/ 28 апреля 2020

Используйте pickle, чтобы превратить модель в байтовый объект: 

model = pickle.dumps(SVR_model)

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