- Add configs for GNN trained at UCL.
- Add "track labels" (
Y_trk_trainin final output file) for GNN training.
- Add option to choose compression for final output files in config (default to
- Add option to choose float precision in final train files. (default to
float16). I didn't add this as an arg to the
TrainSampleWriterclass since it's already carried in the config.
- Write jet label as (5,4,0) in a
flavourfield in the final train files.
- Allow for non chunked storage of final train files.
- Make sure scaling uses
- Some logging improvements to keep track of long steps.
- Ensure out directory exists in resampling stage.
- Increase default chunk size to