Parallel Coordinates Plot in TRAINS

75 views Asked by At

Is there a way to create a parallel coordinates plot in TRAINS (https://github.com/allegroai/trains) package to compare several hyper-parameters in respect to a specific metric?

1

There are 1 answers

0
Martin.B On

Declaimer: I'm a member of TRAINS team

I'm assuming something like https://plot.ly/python/sankey-diagram/#more-complex-sankey-diagram

Unfortunately it is not supported yet, you can always add a request in the GitHub issues.

That said, based on this example You could have 3 axis of hyper-parameter comparison (number_layers, lr_rate, accuracy):

Task.current_task().get_logger().report_scatter3d(
    "performance", "accuracy", iteration=0, 
    mode='markers', scatter=[(number_layers, lr_rate, accuracy)])

Or use the custom column option, and add one the hyper-parameter as an additional column.