Question about the training file

[pipiofzl]

Dear authors,

Thanks for sharing the code, the proposed solution is really inspiring.
I have a question about the training file, after I checked the readme document. Apparaently, there are two files, main.py and main_umil.py. One is used in the training bash file, and the other one is for testing. However, the proposed method like history_score and clustering are not used in main.py while training. Thus, I wonder my observation is correct or not. If so, should I use main_umil.py for training as well?

Best