Не удается развернуть модель, созданную с помощью scikit-learn на ML Engine

Question

Я пытаюсь развернуть модель, созданную с помощью scikit-learn на ML Engine. Я работал над записной книжкой Datalab и после создания модели экспортировал ее в файл, используя joblib.dump (model, 'model.joblib') . Получив этот файл, я скопировал его в облачное хранилище: gsutil cp ./model.joblib gs: //... Наконец, я создал ресурс модели, набрав: gcloud ml-engine models создать модель --regions = us-central1 . Проблема возникает, когда я пытаюсь создать версию модели, так как она говорит, что она ожидает найти файл с сохраненной моделью .pb / .pbtxt. Кто-нибудь знает, как решить эту проблему, не переходя на модель TensorFlow? Любая помощь будет высоко ценится.

Answer 1

Если вы используете Datalab, вы используете старую версию, которая не поддерживает scikit learn.Новое обновление Datalab будет выпущено в любой день.

Однако, обходной путь прост.Просто запустите

%bash
 gcloud components update

в ячейке вашего ноутбука

Answer 2

Вам нужно будет установить несколько флагов, в частности --framework (по умолчанию TensorFlow), но также --runtime-version (должно быть 1.4 или выше) и (возможно) --python-version (по умолчанию 2.7). Попробуйте что-то вроде ( ссылка ):

gcloud beta ml-engine versions create v1 \
  --model my_model \
  --origin gs://path/to/model \
  --runtime-version="1.8" \
  --framework SCIKIT_LEARN
  --python-version="3.5"

Список того, что содержится в каждой версии среды выполнения (т. Е. Чтобы помочь вам выбрать какую), см. здесь .