Я запускаю приведенный ниже код для хранения тегов, а затем для их получения. Как вы можете видеть ниже, Mlflow сохраняет один набор тегов и возвращает другой. импортировать mlflow с mlflow.start_run () как active_run: ...

0
Adeel Hashmi 8 Янв 2021 в 13:09

1 ответ

Лучший ответ

На данный момент вам нужно снова запросить свой запуск в MLflow, чтобы получить запуск со всей информацией, которую вы зарегистрировали. В приведенном ниже примере для этого я вызываю mlflow.get_run(<run_id>).

import mlflow


with mlflow.start_run() as active_run:
  tags = { "my_tag": 1}
  mlflow.set_tags(tags)            
  # Keep track of the run ID of the active run
  run_id = active_run.info.run_id

run = mlflow.get_run(run_id)
print("The tags are ", run.data.tags)
1
Bram 8 Янв 2021 в 17:48