Добавить описание модели при регистрации модели после успешного запуска гипердвигателя

Question

Я успешно обучил модель в службе машинного обучения Azure с использованием Hyperdrive, которая теперь дала экземпляр запуска гипердвигателя

hyperdrive_run = exp.submit(config=hypertune_config)
hyperdrive_run
best_run = hyperdrive_run.get_best_run_by_primary_metric()

В качестве следующего шага я хотел бы зарегистрировать модель при добавлении описания вthe model.:

pumps_rf = best_run.register_model(model_name='pumps_rf', model_path='outputs/rf.pkl')

В разделе "Модели" моего рабочего пространства AML на портале Azure есть столбец description, но метод register_model, похоже, не имеет флага description.Итак, как мне добавить описание к модели, чтобы увидеть его на портале Azure?

Answer 1

Хороший вопрос.,

Однако вы можете обойти это, зарегистрировав модель, используя метод Model.register, который, к счастью, включает аргумент для description, как подробно здесь .В вашем случае вам также необходимо загрузить файлы сначала.

Короче говоря, используйте что-то вроде:

best_run.download_file('outputs/rf.pkl', output_file_path='./rf.pkl')

Model.register(workspace=ws, model_path='./rf.pkl', model_name="pumps_rf", description="There are many models like it, but this one is mine.")