Невозможно загрузить предварительно обученную модель на платформу AI

Я обучил классификатор случайного леса и сохранил модель в виде файла joblib в хранилище Google. Платформа ai выдает ошибку, когда версия создается с использованием среды scikit-learn с этим файлом model.joblib. Ошибка выглядит так:

Create Version failed. Bad model detected with error: "Failed to load model: Could not load the model: /tmp/model/a0001/model.joblib. 0. (Error code: 0)"

Размер модели 283 Мб - вполне допустимый размер.


google-cloud-platform gcp-ai-platform-training scikit-learn joblib random-forest
person Ajay Deshpande    schedule 28.02.2020    source источник

Ответы (1)


arrow_upward
1
arrow_downward

Эта ошибка, как показано в руководстве по устранению неполадок, обычно означает для экспорта модели использовалась неправильная библиотека. Чтобы исправить это, повторно экспортируйте модель, используя правильную библиотеку.

Помните, что библиотека joblib в AI Platform импортируется следующим образом:

from sklearn.externals import joblib

У вас есть пример того, как его экспортировать здесь .

person aemon4    schedule 28.02.2020
comment
Я импортировал joblib как import joblib. Пакет sklearn.externals joblib показывал предупреждение о том, что он может устареть, поэтому я использую прямой импорт — import joblib - person Ajay Deshpande; 28.02.2020
comment
Пробовал использовать sklearn.externals import joblib. Это все еще дает ту же ошибку. ОШИБКА: не удалось создать версию. Неверная модель обнаружена с ошибкой: Не удалось загрузить модель: Не удалось загрузить модель: /tmp/model/0001/model.joblib. 0. (Код ошибки: 0) - person Ajay Deshpande; 28.02.2020
comment
Если проблема с библиотекой joblib по-прежнему возникает, попробуйте использовать библиотеку pickle и сохраните ее как model.pkl. - person aemon4; 02.03.2020