Проблема при загрузке предварительно обученной модели - PullRequest
0 голосов
/ 10 ноября 2019

Я обучил свою модель с помощью Google Colab и сохранил ее как model.pkl. Когда я пытаюсь загрузить модель в свой ноутбук, она выдает следующую ошибку:

Traceback (most recent call last):
File "app.py", line 8, in <module>
model = pickle.load(open('model.pkl', 'rb'))
File "sklearn\tree\_tree.pyx", line 606, in sklearn.tree._tree.Tree.__cinit__
ValueError: Buffer dtype mismatch, expected 'SIZE_t' but got 'long long'

Я провел некоторое исследование по вышеуказанной ошибке и узнал, что код случайного леса использует разные типы для индексов. на 32-битных и 64-битных машинах. Я видел похожий вопрос на этой платформе, но НЕ удовлетворен принятым ответом, потому что ответ, предлагающий обучить модель снова, не подходит в случае, так как есть много вещей, которые нужно переделать, и я нене хотите снова загружать сервер.

Есть предложения или решения?

1 Ответ

0 голосов
/ 11 ноября 2019

Не уверен насчет формата '.pkl', но вы можете попробовать сохранить его как
model.save ('modelweight.h5') и затем загрузить как model.load ('modelweight.h5'). Это должно работать. Благодаря.

...