Не удается развернуть обученную модель с помощью пользовательской процедуры прогнозирования.Сбой с ошибкой загрузки модели - PullRequest
0 голосов
/ 15 мая 2019

Я пытаюсь развернуть пользовательскую процедуру прогнозирования, текст этой категории. Я обучил модель и предоставил .pkl в каталоге модели облачного хранилища. Сбой создания версии модели с:

ERROR: (gcloud.beta.ai-platform.versions.create) Bad model detected with error:  "Failed to load model: User-provided package text_categorizer-0.2.tar.gz failed to install: Command '
['python-default', '-m', 'pip', 'install', '--target=/tmp/custom_lib', '--no-cache-dir', '-b', '/tmp/pip_builds', '/tmp/custom_code/text_categorizer-0.2.tar.gz']' returned non-zero e
xit status 1 (Error code: 0)"

Также есть какие-нибудь журналы, которые я мог бы проверить, чтобы узнать, почему модель не может быть загружена? Все классы Predictor доступны в файле .tar.gz, загруженном в облачное хранилище. Убедитесь, что корзина находится в том же регионе, что и созданная модель. Файлы модели находятся в каталоге с названием model, а файл tar.gz размещен в главном каталоге с именем textcateg. Файл tar.gz содержит класс Python предиктора.

Команда для создания версии:

gcloud beta ai-platform versions create ver3 --model text_categorizer --runtime-version 1.13 --python-version 3.5 --origin gs://mdmceml-bucket/textcateg/model/ --package-uris gs://mdmceml-bucket/textcateg/text_categorizer-0.2.tar.gz --prediction-class predictor.GCPPredictor

Соответствующий фрагмент кода от GCPPredictor ::

    def __init__(self, le, clf):
        """Stores artifacts for prediction. Only initialized via `from_path`.
        """
        self._le = le
        self._clf = clf

    @classmethod
    def from_path(cls, model_dir):
        """Creates an instance of Predictor using the given path.

        Loading of the predictor should be done in this method.

        Args:
            model_dir: The local directory that contains the exported model
                file along with any additional files uploaded when creating the
                version resource.

        Returns:
            An instance implementing this Predictor class.
        """

        model_path = os.path.join(model_dir, 'category_classifier.pkl')
        with open(model_path, 'rb') as f:
            model = pickle.load(f)

        le_path = os.path.join(model_dir, 'labelencoder.pkl')
        with open(le_path, 'rb') as f1:
            le = pickle.load(f1)

        return cls(le, model)

Создание версии модели завершается неудачно с:

ERROR: (gcloud.beta.ai-platform.versions.create) Bad model detected with error:  "Failed to load model: User-provided package text_categorizer-0.2.tar.gz failed to install: Command '
['python-default', '-m', 'pip', 'install', '--target=/tmp/custom_lib', '--no-cache-dir', '-b', '/tmp/pip_builds', '/tmp/custom_code/text_categorizer-0.2.tar.gz']' returned non-zero e
xit status 1 (Error code: 0)"
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...