Блог

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

Мы регистрируем несколько показателей, таких как потери при обучении и проверке, а также точность в разные эпохи. Только это 5 показателей, по всем из которых я хотел бы получить полную историю.

Вот примерный пример:

 from mlflow.tracking import MlflowClient

client = MLFlowClient()

epochs = client.get_metric_history(id, 'epochs')
train_loss = client.get_metric_history(id, 'train_loss')
val_loss = client.get_metric_history(id, 'val_loss')
train_acc = client.get_metric_history(id, 'train_acc')
val_acc = client.get_metric_history(id, 'val_acc')
 

Каждая отдельная строка здесь представляет собой отдельный вызов, и вся эта последовательность в конечном итоге занимает некоторое время. Есть ли способ, которого мне не хватает, чтобы получить историю показателей для нескольких (или всех) показателей одновременно?
Выборка client.get_run(id).data.metrics показывает только конечное значение, в то время как мне нужна вся история.