Заявитель: я являюсь членом команды TRAINS
Я предполагаю, что-то вроде https://plot.ly/python/sankey-diagram/#more-complex-sankey-diagram
К сожалению, пока это не поддерживается, вы всегда можете добавить запрос в GitHubпроблемы.
Тем не менее, основываясь на этом примере Вы можете иметь 3 оси сравнения гиперпараметров (number_layers, lr_rate, точность):
Task.current_task().get_logger().report_scatter3d(
"performance", "accuracy", iteration=0,
mode='markers', scatter=[(number_layers, lr_rate, accuracy)])
Или использовать настраиваемый столбец и добавьте один гиперпараметр в качестве дополнительного столбца.