Есть ли способ экспортировать или просмотреть классификатор, созданный в Sagemaker, чтобы мы могли видеть, какие веса / константы используются при оценке модели - PullRequest
3 голосов
/ 03 июля 2019

Я создал простую линейную модель ученика, используя sagemaker, и, хотя я могу развернуть ее в наборе тестовых данных, я хотел бы иметь возможность получить фактическое уравнение, которое модель использует для классификации значений (то есть для линейной регрессии это уравнениелинии).

1 Ответ

1 голос
/ 03 июля 2019

Вы можете открыть артефакт модели с помощью mxnet и просмотреть вес и смещение - см. Код ниже, вставленный из этого сообщения на форуме

import os
import mxnet as mx
import boto3

bucket = "<your_bucket"
key = "<your_model_prefix>"
boto3.resource('s3').Bucket(bucket).download_file(key, 'model.tar.gz')

os.system('tar -zxvf model.tar.gz')

# Linear learner model is itself a zip file, containing a mxnet model and other metadata.
# First unzip the model.
os.system('unzip model_algo-1') 

# Load the mxnet module
mod = mx.module.Module.load("mx-mod", 0)

# model's weights
mod._arg_params['fc0_weight'].asnumpy().flatten()

# model bias
mod._arg_params['fc0_bias'].asnumpy().flatten()
...