【问题标题】:Gcloud ai-platform, can't create model with own prediction-classGcloud ai-platform,无法使用自己的预测类创建模型
【发布时间】:2019-10-03 14:43:48
【问题描述】:

我尝试关注 AI Platform tutorial 上传模型和预测例程,但有一部分失败,我不明白为什么。

我的预测类和他们教程中的一样:

%%writefile predictor.py
import os
import pickle

import numpy as np
from sklearn.datasets import load_iris
from sklearn.externals import joblib

class MyPredictor(object):
  def __init__(self, model, preprocessor):
    self._model = model
    self._preprocessor = preprocessor
    self._class_names = load_iris().target_names

  def predict(self, instances, **kwargs):
    inputs = np.asarray(instances)
    preprocessed_inputs = self._preprocessor.preprocess(inputs)
    if kwargs.get('probabilities'):
      probabilities = self._model.predict_proba(preprocessed_inputs)
      return probabilities.tolist()
    else:
      outputs = self._model.predict(preprocessed_inputs)
      return [self._class_names[class_num] for class_num in outputs]

  @classmethod
  def from_path(cls, model_dir):
    model_path = os.path.join(model_dir, 'model.joblib')
    model = joblib.load(model_path)

    preprocessor_path = os.path.join(model_dir, 'preprocessor.pkl')
    with open(preprocessor_path, 'rb') as f:
      preprocessor = pickle.load(f)

    return cls(model, preprocessor)

我用来在云端创建模型的代码是:

! gcloud beta ai-platform versions create {VERSION_NAME} \
  --model {MODEL_NAME} \
  --runtime-version 1.13 \
  --python-version 3.5 \
  --origin gs://{BUCKET_NAME}/custom_prediction_routine_tutorial/model/ \
  --package-uris gs://{BUCKET_NAME}/custom_prediction_routine_tutorial/my_custom_code-0.1.tar.gz \
  --prediction-class predictor.MyPredictor

但我最终遇到了这样一个奇怪的错误:

ERROR: (gcloud.beta.ai-platform.versions.create) Bad model detected with error:  "Failed to load model: Unexpected error when loading the model: 'ascii' codec can't decode byte 0xf9 in position 2: ordinal not in range(128) (Error code: 0)"

问题是当我运行相同的命令时没有:

--prediction-class predictor.MyPredictor

它工作正常。

有人知道这是什么原因吗?我认为model.joblib 可能有编码问题,但是当我自己加载它时没有任何问题

【问题讨论】:

  • 重复了教程中的步骤,可以确认教程中存在问题。在 GitHub 上创建了一个问题:github.com/GoogleCloudPlatform/cloudml-samples/issues/419
  • @Hadrien Berthier 你能再试一次吗?
  • 我已经设法让它工作了,问题是在教程中他们使用pickle作为预处理器,使用Joblib作为模型,你需要同时使用Joblib。

标签: gcloud google-cloud-ml


【解决方案1】:

我找到了解决办法,

在教程中,他们使用 pickle 来保存创建的预处理器对象,并使用 Joblib 来保存模型。

您需要使用 Joblib 保存两者,然后将其发送到谷歌存储。

【讨论】:

    猜你喜欢
    • 2021-04-23
    • 2020-01-30
    • 2021-08-28
    • 1970-01-01
    • 1970-01-01
    • 2021-05-10
    • 1970-01-01
    • 1970-01-01
    • 2020-11-21
    相关资源
    最近更新 更多