Как запустить веб-интерфейс mlflow из Amazon SageMaker?

Question

Я хочу использовать веб-интерфейс mlflow из записной книжки на Amazon SageMaker. Но данный адрес http://127.0.0.1:5000/ не работает.

Я установил mlflow на ноутбук SageMaker.

Этот код работает хорошо:

import mlflow
mlflow.start_run()
mlflow.log_param("my", "param")
mlflow.log_metric("score", 100)
mlflow.end_run()

И тогда, если я бегу

! mlflow ui

Я получаю ожидаемый результат:

[2019-04-09 11:15:52 +0000] [17980] [INFO] Starting gunicorn 19.9.0
[2019-04-09 11:15:52 +0000] [17980] [INFO] Listening at: http://127.0.0.1:5000 (17980)
[2019-04-09 11:15:52 +0000] [17980] [INFO] Using worker: sync
[2019-04-09 11:15:52 +0000] [17983] [INFO] Booting worker with pid: 17983

Однако после этого при переходе http://127.0.0.1:5000 в моем браузере ничего не загружается.

Я думаю, что это 127.0.0.1 не правильный адрес, но как я могу узнать, какой адрес использовать вместо этого?

Answer 1

Привет и спасибо за использование SageMaker!

К сожалению, в настоящее время mlflow не совместим с SageMaker. Мы предоставляем функцию, которая может поддерживать эти сценарии. SageMaker включает в себя плагин, называемый jupyter-server-proxy, который позволяет размещать другие веб-приложения на вашем экземпляре SageMaker Notebook, например TensorBoard.

В случае с mlflow, я почти смог заставить ваш пример работать, посетив https://mynotebookinstance.notebook.us-east-1.sagemaker.aws/proxy/5000/ (обратите внимание, как номер порта перемещается в конец), но, к сожалению, mlflow отображает ошибку, поскольку в настоящее время предполагается, что он не работает по корневому URL-пути.

Я создал проблему в репозитории mlflow GitHub: https://github.com/mlflow/mlflow/issues/1120 Пожалуйста, отметьте эту проблему, чтобы быть в курсе ее статуса.

Бест, Кевин

Answer 2

Я мог бы решить проблему, перенаправив порт вместо использования прокси-ссылки.

В локальном терминале выполните следующую команду:

      ssh -N -L 5000:127.0.0.1:5000 ec2-user@your_sagemaker_instance_ip_address 

А затем просто зайдите по адресу http://127.0.0.1:5000 .