Как установить имя_процесса в командной строке mlflow? - PullRequest
Новая
       14

Как установить имя_процесса в командной строке mlflow?

1 голос
/ 02 ноября 2019

MLFlow версия: 1.4.0 Версия Python: 3.7.4

Я использую пользовательский интерфейс как mlflow server... со всеми необходимыми параметрами командной строки.

Я регистрируюсь в MLFlow как проект MLFlow с соответствующим файлом MLproject.yaml. Проект выполняется в контейнере Docker, поэтому CMD выглядит следующим образом:

mlflow run . -P document_ids=${D2V_DOC_IDS} -P corpus_path=... --no-conda --experiment-name=${EXPERIMENT_NAME}

Запуск эксперимента, подобного этому, приводит к пустому run_name. Я знаю, что есть run_id, но я также хотел бы увидеть имя run_name и установить его в своем коде - либо в командной строке, либо в моем коде как mlflow.log.....

Я смотрел на Можно ли установить / изменить имя прогона mlflow после первоначального создания прогона? , но я хочу программно установить имя прогона вместо его изменения вручную в пользовательском интерфейсе.

1 Ответ

1 голос
/ 10 ноября 2019

Одним из параметров для mlflow.start_run() является run_name. Это даст вам программный доступ для установки имени прогона для каждой итерации. Смотри документы здесь .

Вот пример: 

from datetime import datetime

## Define the name of our run
name = "this run is gonna be bananas" + datetime.now()

## Start a new mlflow run and set the run name
with mlflow.start_run(run_name = name):

    ## ...train model, log metrics/params/model...

    ## End the run
    mlflow.end_run()

Если вы хотите включить набор имени как часть проекта MLflow, вам нужно будет указать его в качестве параметра в точках входа впроект. Он находится в файле MLproject . Затем вы можете передать эти значения в функцию mlflow.start_run() из командной строки.

...