在AWS中运行mlflow ui

Question

我想在sagemaker中运行mlflow UI，但是它根本不工作，当它输出到它的http地址时，会导致“无法到达这个站点”。

以下是代码：

def mlflow_test(server_uri, experiment_name):
    mlflow.set_tracking_uri(server_uri)
    mlflow.set_experiment(experiment_name)
    with mlflow.start_run():
        params = {
            "n-estimators": 100,
            "min-samples-leaf": 10,
            "features": 'feature_test'
        }
        mlflow.log_params(params)
        mlflow.log_metric('foo', 5)
        mlflow.end_run()

运行该代码将返回：

[2022-05-24 15:48:44 +0000] [27820] [INFO] Starting gunicorn 20.1.0
[2022-05-24 15:48:44 +0000] [27820] [INFO] Listening at: http://127.0.0.1:5000 (27820)
[2022-05-24 15:48:44 +0000] [27820] [INFO] Using worker: sync
[2022-05-24 15:48:44 +0000] [27823] [INFO] Booting worker with pid: 27823

去http://127.0.0.1:5000链接是行不通的。有人知道如何让mlflow ui在sagemaker中运行吗？没有太多的信息，这是在一个容易理解的水平。我只想记录我的指标和参数，并使用mlflow ui查看它们。

Answer 1

您是在SageMaker Studio中工作还是在经典笔记本实例中工作？从技术上讲，您可以使用Studio终端尝试启动MLflow，但我不建议这样做。最好在EC2上设置这样的东西，在这里您可以完全控制环境的设置。