About the train and test files

[zhangzhengde0225]

Dear Author: Thank you for your project. I'm a ZhengDe ZHang, an A.P. from IHEP CAS. I'm trying to reproduce your code. Could you please tell me where can I find the train and test files (or examples files) needed in trian.py "/path/to/train_files/////output_*.root"

[hqucms]

Hi @zhangzhengde0225 -- sorry for the late reply, somehow I missed this message. You can find an example root file here. This example file is actually extracted from our new JetClass dataset, and you can find more information and the full set of code and instructions in https://github.com/jet-universe/particle_transformer.