Could you kindly explain the pickle format used in your input dataset?

[bgeneto]

For example, in the DW we have the following shape for all_data:

# dataset = 'DW_1000T_1x.pk'
print(all_data.shape)
(160, 9999, 6)

I'm supposing (in this case) we have:

• 160 different simulations (with different physical parameters).
• 9999 points
• and the last (third) dimension is unknown to me. Maybe index 0 is time and the last one is the actual solution. What about the others?

TIA.